From 5f7ffbc2d6bc233539414e58e687f33d71889a55 Mon Sep 17 00:00:00 2001 From: Qdriving <35327911+Qdriving@users.noreply.github.com> Date: Mon, 28 Sep 2020 12:37:42 +0800 Subject: [PATCH] Update reader.py --- models/rank/dien/reader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/rank/dien/reader.py b/models/rank/dien/reader.py index b79838c8..56c8c18c 100755 --- a/models/rank/dien/reader.py +++ b/models/rank/dien/reader.py @@ -102,9 +102,9 @@ class Reader(ReaderBase): else: len_seq = len(b[i][0]) start_idx = random.randint(0, self.max_neg_item - len_seq - 1) - self.neg_candidate_item[start_idx:start_idx + len_seq +1] = b[ + self.neg_candidate_item[start_idx:start_idx + len_seq + 1] = b[ i][0] - self.neg_candidate_cat[start_idx:start_idx + len_seq +1] = b[ + self.neg_candidate_cat[start_idx:start_idx + len_seq + 1] = b[ i][1] for _ in range(len(b[i][0])): -- GitLab