diff --git a/.gitignore b/.gitignore index 35ab955..ed1bb18 100644 --- a/.gitignore +++ b/.gitignore @@ -16,3 +16,4 @@ saved/ *.egg-info/ docs/build/ wandb/* +.DS_Store \ No newline at end of file diff --git a/recbole_cdr/sampler/crossdomain_sampler.py b/recbole_cdr/sampler/crossdomain_sampler.py index a8c406a..5c08ccd 100644 --- 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