From 34cf73f8f5703bfbfbc4107755538fa2cc0eb619 Mon Sep 17 00:00:00 2001 From: yinhaofeng <1841837261@qq.com> Date: Tue, 12 Apr 2022 14:18:33 +0000 Subject: [PATCH] mind_loss --- models/recall/mind/mind_reader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/recall/mind/mind_reader.py b/models/recall/mind/mind_reader.py index ef211e275..0d1b78df4 100644 --- a/models/recall/mind/mind_reader.py +++ b/models/recall/mind/mind_reader.py @@ -61,6 +61,7 @@ def __iter__(self): item_list = self.graph[user_id] if len(item_list) <= 4: continue + random.seed(12345) k = random.choice(range(4, len(item_list))) item_id = item_list[k]