Skip to content

Commit 1f1d93e

Browse files
authored
Merge pull request #711 from wangzhen38/fix_bst
fix bst bug from issue
2 parents eff6da2 + 2ce9739 commit 1f1d93e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/rank/bst/net.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -433,9 +433,9 @@ def forward(self, userid, hist_item_seq, hist_cat_seq, position_seq,
433433
target_position_emb = self.target_position_emb_attr(target_position)
434434

435435
item_sequence = paddle.concat(
436-
[hist_item_emb, hist_item_emb, hist_position_emb], axis=2)
436+
[hist_item_emb, hist_cat_emb, hist_position_emb], axis=2)
437437
target_sequence = paddle.concat(
438-
[target_item_emb, target_item_emb, target_position_emb], axis=2)
438+
[target_item_emb, target_cat_emb, target_position_emb], axis=2)
439439

440440
#print(position_sequence_target.shape)
441441
whole_embedding = paddle.concat(

0 commit comments

Comments
 (0)