Merge pull request #1301 from chenyushuo/master

FIX: bug fix for unused_col is not used when using benchmark_file.
RUCAIBox · Jun 3, 2022 · ddff621 · ddff621
2 parents 959ec8b + 555a8bd
commit ddff621
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/recbole/data/dataset/dataset.py b/recbole/data/dataset/dataset.py
@@ -1474,6 +1474,7 @@ def build(self):
         self._change_feat_format()
 
         if self.benchmark_filename_list is not None:
+            self._drop_unused_col()
             cumsum = list(np.cumsum(self.file_size_list))
             datasets = [self.copy(self.inter_feat[start:end]) for start, end in zip([0] + cumsum[:-1], cumsum)]
             return datasets