def __init__(self, num_users, num_items, D=10, Dprime=60, hidden_units_per_layer=50,
latent_normal_init_params={'mean': 0.0, 'stddev': 0.1}, model_filename='model/nnmf.ckpt'):
self.num_users = num_users
self.num_items = num_items
self.D = D
self.Dprime = Dprime
self.hidden_units_per_layer = hidden_units_per_layer
self.latent_normal_init_params = latent_normal_init_params
self.model_filename = model_filename
# Internal counter to keep track of current iteration
self._iters = 0
# Input
self.user_index = tf.placeholder(tf.int32, [None])
self.item_index = tf.placeholder(tf.int32, [None])
self.r_target = tf.placeholder(tf.float32, [None])
# Call methods to initialize variables and operations (to be implemented by children)
self._init_vars()
self._init_ops()
# RMSE
self.rmse = tf.sqrt(tf.reduce_mean(tf.square(tf.sub(self.r, self.r_target))))
评论列表
文章目录