ATRank
ATRank copied to clipboard
An Attention-Based User Behavior Modeling Framework for Recommendation
hi,dear 大佬,1.14或者1.15的版本能用吗?
Traceback (most recent call last): File "train.py", line 184, in tf.app.run() File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/platform/app.py", line 40, in run _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef) File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 299, in run _run_main(main, args) File "/usr/local/lib/python3.6/dist-packages/absl/app.py",...
运行rnn、rnn_att、bpr模型时,在 ``` python def train(self, sess, uij, l): 103 loss, _ = sess.run([self.loss, self.train_op], feed_dict={ 104 self.u: uij[0], 105 self.i: uij[1], 106 self.y: uij[2], 107 self.hist_i: uij[3], 108 self.sl: uij[4],...
It seems that you atrank is different from the one described in paper. For example, bilinear attention is used in paper, but scale-dot attention here. Vanilla attention in paper, but...