params.py

# Copyright (c) 2018-present, Royal Bank of Canada.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
#
class Params:

    def __init__(self, 
                 ne=500, 
                 bsize=512, 
                 lr=0.001, 
                 reg_lambda=0.0, 
                 emb_dim=100, 
                 neg_ratio=20, 
                 dropout=0.5,
                 e_epoch = 10,  
                 save_each=10,  
                 se_prop=0.68,
                 alp = 0.5):

        self.ne = ne
        self.bsize = bsize
        self.lr = lr
        self.reg_lambda = reg_lambda
        self.emb_dim = emb_dim
        self.s_emb_dim = int(se_prop*emb_dim)
        self.t_emb_dim = emb_dim - int(se_prop*emb_dim)
        self.save_each = save_each
        self.neg_ratio = neg_ratio
        self.dropout = dropout
        self.e_epoch = e_epoch
        self.se_prop = se_prop
        self.dim_ff = self.t_emb_dim * 2
        self.n_head = 4
        self.alp = alp
        self.d_k = self.t_emb_dim // self.n_head
        self.d_v = self.t_emb_dim // self.n_head
        
    def str_(self):
        return str(self.ne) + "_" + str(self.bsize) + "_" + str(self.lr) + "_" + str(self.reg_lambda) + "_" + str(self.s_emb_dim) + "_" + str(self.neg_ratio) + "_" + str(self.dropout) + "_" + str(self.t_emb_dim) + "_" + str(self.save_each) + "_" + str(self.se_prop)