Merge pull request #58 from WYH-han/bug/sampler

Bug of sampler
RUCAIBox · Dec 8, 2023 · d339918 · d339918
2 parents 5bb219d + 2c7470d
commit d339918
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -16,3 +16,4 @@ saved/
 *.egg-info/
 docs/build/
 wandb/*
+.DS_Store
diff --git a/recbole_cdr/sampler/crossdomain_sampler.py b/recbole_cdr/sampler/crossdomain_sampler.py
@@ -195,8 +195,8 @@ def __init__(self, phases, dataset, built_datasets, distribution='uniform'):
         self.uid_field = self.dataset.uid_field
         self.iid_field = self.dataset.iid_field
 
-        self.overlapped_item_num = dataset.num_overlap_item + 1  # for [pad]
-        self.overlapped_user_num = dataset.num_overlap_user + 1
+        self.overlapped_item_num = dataset.num_overlap_item
+        self.overlapped_user_num = dataset.num_overlap_user
 
         self.source_only_item_num = dataset.num_source_only_item
         self.source_only_user_num = dataset.num_source_only_user