Bugfix andreeaiana#3

Poseidondon · Jun 11, 2024 · 848a7bb · 848a7bb
1 parent 8caa65b
commit 848a7bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/configs/data/rumind_rec.yaml b/configs/data/rumind_rec.yaml
@@ -52,7 +52,7 @@ entity_conf_threshold: 0.5
 
 sentiment_annotator: null
 
-valid_time_split: "2024-05-29 22:32:51" # interactions with timestamp before this data will be used for training
+valid_time_split: "2024-05-31 22:14:05" # interactions with timestamp before this data will be used for training
 
 # dataset settings
 neg_sampling_ratio: 4

diff --git a/newsreclib/data/components/rec_dataset.py b/newsreclib/data/components/rec_dataset.py
@@ -47,11 +47,7 @@ def __getitem__(
         candidates = np.array(bhv["candidates"])
         labels = np.array(bhv["labels"])
 
-        try:
-            candidates, labels = self._sample_candidates(candidates, labels)
-        except Exception:
-            print('Candidates:', candidates)
-            print('Labels:', labels)
+        candidates, labels = self._sample_candidates(candidates, labels)
 
         history = self.news.loc[history]
         candidates = self.news.loc[candidates]