def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_T_SeqRec,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
item_history_embedding_new = tf.concat([
self.item_history_embedding,
tf.expand_dims(self.timelast_history, -1)
], -1)
item_history_embedding_new = tf.concat([
item_history_embedding_new,
tf.expand_dims(self.timenow_history, -1)
], -1)
# RNN layer(1)
with tf.name_scope('rnn'):
rnn_outputs, final_state = dynamic_rnn(
Time4LSTMCell(HIDDEN_SIZE),
inputs=item_history_embedding_new,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="time4lstm")
tf.summary.histogram('LSTM_outputs', rnn_outputs)
last_inps = tf.concat([self.target_item_embedding, final_state[1]], 1)
self.fcn_net(last_inps, use_dice=True)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_SLi_Rec_Fixed,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
# Attention layer(1)
with tf.name_scope('Attention_layer_1'):
att_outputs1, alphas1 = attention_HAN(
self.item_history_embedding,
attention_size=ATTENTION_SIZE,
return_alphas=True)
att_fea1 = tf.reduce_sum(att_outputs1, 1)
tf.summary.histogram('att_fea1', att_fea1)
item_history_embedding_new = tf.concat([
self.item_history_embedding,
tf.expand_dims(self.timelast_history, -1)
], -1)
item_history_embedding_new = tf.concat([
item_history_embedding_new,
tf.expand_dims(self.timenow_history, -1)
], -1)
# RNN layer
with tf.name_scope('rnn'):
rnn_outputs, final_state1 = dynamic_rnn(
Time4LSTMCell(HIDDEN_SIZE),
inputs=item_history_embedding_new,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="time4lstm")
tf.summary.histogram('LSTM_outputs', rnn_outputs)
# Attention layer(2)
with tf.name_scope('Attention_layer_2'):
att_outputs2, alphas2 = attention_FCN(self.target_item_embedding,
rnn_outputs,
ATTENTION_SIZE,
self.mask,
softmax_stag=1,
stag='1_1',
mode='LIST',
return_alphas=True)
tf.summary.histogram('alpha_outputs2', alphas2)
att_fea2 = tf.reduce_sum(att_outputs2, 1)
tf.summary.histogram('att_fea2', att_fea2)
# alpha
with tf.name_scope('User_alpha'):
user_alpha = 0.2
user_embed = att_fea1 * user_alpha + att_fea2 * (1.0 - user_alpha)
last_inps = tf.concat([self.target_item_embedding, user_embed], 1)
self.fcn_net(last_inps, use_dice=True)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_NARM,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
# RNN layer(1)
with tf.name_scope('rnn_1'):
rnn_outputs1, final_state1 = dynamic_rnn(
LSTMCell(HIDDEN_SIZE),
inputs=self.item_history_embedding,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="lstm_1")
tf.summary.histogram('LSTM_outputs1', rnn_outputs1)
# RNN layer(2)
with tf.name_scope('rnn_2'):
rnn_outputs2, final_state2 = dynamic_rnn(
LSTMCell(HIDDEN_SIZE),
inputs=self.item_history_embedding,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="lstm_2")
tf.summary.histogram('LSTM_outputs2', rnn_outputs2)
# Attention layer
with tf.name_scope('Attention_layer'):
att_outputs, alphas = attention_FCN(final_state1[1],
rnn_outputs2,
ATTENTION_SIZE,
self.mask,
softmax_stag=1,
stag='1_1',
mode='LIST',
return_alphas=True)
tf.summary.histogram('alpha_outputs', alphas)
att_fea = tf.reduce_sum(att_outputs, 1)
last_inps = tf.concat(
[final_state1[1], att_fea, self.target_item_embedding], 1)
self.fcn_net(last_inps, use_dice=True)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_DIEN,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
# RNN layer(1)
with tf.name_scope('rnn_1'):
rnn_outputs, _ = dynamic_rnn(GRUCell(HIDDEN_SIZE),
inputs=self.item_history_embedding,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="gru_1")
tf.summary.histogram('GRU_outputs', rnn_outputs)
# Attention layer
with tf.name_scope('Attention_layer'):
att_outputs, alphas = attention_FCN(self.target_item_embedding,
rnn_outputs,
ATTENTION_SIZE,
self.mask,
softmax_stag=1,
stag='1_1',
mode='LIST',
return_alphas=True)
tf.summary.histogram('alpha_outputs', alphas)
# RNN layer(2)
with tf.name_scope('rnn_2'):
rnn_outputs2, final_state2 = dynamic_rnn(
VecAttGRUCell(HIDDEN_SIZE),
inputs=rnn_outputs,
att_scores=tf.expand_dims(alphas, -1),
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="gru_2")
tf.summary.histogram('GRU2_Final_State', final_state2)
last_inps = tf.concat([self.target_item_embedding, final_state2], 1)
self.fcn_net(last_inps, use_dice=True)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_LSTMPP,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
# Attention layer
with tf.name_scope('Attention_layer'):
att_outputs, alphas = attention_HAN(self.item_history_embedding,
attention_size=ATTENTION_SIZE,
return_alphas=True)
att_fea = tf.reduce_sum(att_outputs, 1)
tf.summary.histogram('att_fea', att_fea)
# RNN layer
with tf.name_scope('rnn'):
rnn_outputs, final_state = dynamic_rnn(
LSTMCell(HIDDEN_SIZE),
inputs=self.item_history_embedding,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="lstm")
tf.summary.histogram('LSTM_outputs', rnn_outputs)
# alpha
with tf.name_scope('User_alpha'):
concat_all = tf.concat([
self.target_item_embedding, att_fea, final_state[1],
tf.expand_dims(self.timenow_history[:, -1], -1)
], 1)
concat_att1 = tf.layers.dense(concat_all,
80,
activation=tf.nn.sigmoid,
name='concat_att1')
concat_att2 = tf.layers.dense(concat_att1,
40,
activation=tf.nn.sigmoid,
name='concat_att2')
user_alpha = tf.layers.dense(concat_att2,
1,
activation=tf.nn.sigmoid,
name='concat_att3')
user_embed = att_fea * user_alpha + final_state[1] * (1.0 -
user_alpha)
last_inps = tf.concat([self.target_item_embedding, user_embed], 1)
self.fcn_net(last_inps, use_dice=True)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_LSTM,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
# RNN layer
with tf.name_scope('rnn'):
rnn_outputs, final_state = dynamic_rnn(
LSTMCell(HIDDEN_SIZE),
inputs=self.item_history_embedding,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="lstm")
tf.summary.histogram('LSTM_outputs', rnn_outputs)
last_inps = tf.concat([self.target_item_embedding, final_state[1]], 1)
self.fcn_net(last_inps, use_dice=False)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_TC_SeqRec,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
item_history_embedding_new = tf.concat([
self.item_history_embedding,
tf.expand_dims(self.timelast_history, -1)
], -1)
item_history_embedding_new = tf.concat([
item_history_embedding_new,
tf.expand_dims(self.timenow_history, -1)
], -1)
# RNN layer
with tf.name_scope('rnn'):
rnn_outputs, final_state = dynamic_rnn(
Time4LSTMCell(HIDDEN_SIZE),
inputs=item_history_embedding_new,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="time4lstm")
tf.summary.histogram('LSTM_outputs', rnn_outputs)
# Attention layer
with tf.name_scope('Attention_layer'):
att_outputs, alphas = attention_FCN(self.target_item_embedding,
rnn_outputs,
ATTENTION_SIZE,
self.mask,
softmax_stag=1,
stag='1_1',
mode='LIST',
return_alphas=True)
tf.summary.histogram('alpha_outputs', alphas)
att_fea = tf.reduce_sum(att_outputs, 1)
last_inps = tf.concat([self.target_item_embedding, att_fea], 1)
self.fcn_net(last_inps, use_dice=True)
def __init__(self, user_number, item_number, cate_number, EMBEDDING_DIM,
HIDDEN_SIZE, ATTENTION_SIZE):
super(Model_CARNN,
self).__init__(user_number, item_number, cate_number,
EMBEDDING_DIM, HIDDEN_SIZE, ATTENTION_SIZE)
self.item_history_embedding = tf.concat([
self.item_history_embedding,
tf.expand_dims(self.timeinterval_history, -1)
], -1)
# RNN layer
with tf.name_scope('rnn'):
rnn_outputs, final_state = dynamic_rnn(
CARNNCell(HIDDEN_SIZE),
inputs=self.item_history_embedding,
sequence_length=self.sequence_length,
dtype=tf.float32,
scope="carnn")
tf.summary.histogram('CARNN_outputs', rnn_outputs)
last_inps = tf.concat([final_state, self.target_item_embedding], 1)
self.fcn_net(last_inps, use_dice=True)
