Merge pull request #2054 from thaiminhpv/thaiminhpv/correct-MIND-user…

…-behavior-history-construction Correct MIND user behavior history construction
recommenders-team · Jan 8, 2024 · e3e3ee7 · e3e3ee7
2 parents 4e9a546 + 08dc249
commit e3e3ee7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/recommenders/models/newsrec/io/mind_all_iterator.py b/recommenders/models/newsrec/io/mind_all_iterator.py
@@ -146,7 +146,7 @@ def init_behaviors(self, behaviors_file):
 
                 history = [self.nid2index[i] for i in history.split()]
                 history = [0] * (self.his_size - len(history)) + history[
-                    : self.his_size
+                    -self.his_size :
                 ]
 
                 impr_news = [self.nid2index[i.split("-")[0]] for i in impr.split()]

diff --git a/recommenders/models/newsrec/io/mind_iterator.py b/recommenders/models/newsrec/io/mind_iterator.py
@@ -119,7 +119,7 @@ def init_behaviors(self, behaviors_file):
 
                 history = [self.nid2index[i] for i in history.split()]
                 history = [0] * (self.his_size - len(history)) + history[
-                    : self.his_size
+                    -self.his_size :
                 ]
 
                 impr_news = [self.nid2index[i.split("-")[0]] for i in impr.split()]