def __init__(self, sess, num_user, num_item,
hidden_encoder_dim=216, hidden_decoder_dim=216, latent_dim=24,
learning_rate=0.002, batch_size=64, reg_param=0,
user_embed_dim=216, item_embed_dim=216, activate_fn=tf.tanh, vae=True):
if reg_param < 0 or reg_param > 1:
raise ValueError("regularization parameter must be in [0,1]")
self.sess = sess
self.num_user = num_user
self.num_item = num_item
self.hidden_encoder_dim = hidden_encoder_dim
self.hidden_decoder_dim = hidden_decoder_dim
self.latent_dim = latent_dim
self.learning_rate = learning_rate
self.batch_size = batch_size
self.reg_param = reg_param
self.user_embed_dim = user_embed_dim
self.item_embed_dim = item_embed_dim
self.activate_fn = activate_fn
self.vae = vae
self.build_model()
评论列表
文章目录