jinze1994 / ATRank

An Attention-Based User Behavior Modeling Framework for Recommendation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ValueError: setting an array element with a sequence.

junthy-me opened this issue · comments

运行rnn、rnn_att、bpr模型时,在

       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],
109         self.lr: l,
110         })
111     return loss
       def train(self, sess, uij, l):
113     loss, _ = sess.run([self.loss, self.train_op], feed_dict={
114         self.u: uij[0],
115         self.i: uij[1],
116         self.y: uij[2],
117         self.hist_i: uij[2],
118         self.sl: uij[4],
119         self.lr: l,
120         })
121     return loss

都遇到了这个问题,希望作者能够更新代码解决这个问题。