Update dataset.py

RUCAIBox · Mar 8, 2023 · 89be22c · 89be22c
1 parent 1758db4
commit 89be22c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole_cdr/data/dataset.py b/recbole_cdr/data/dataset.py
@@ -119,7 +119,7 @@ def _remap_fields(self, field_names, map_dict):
             if self.user_feat is not None and field_name in self.user_feat.columns:
                 self.user_feat[field_name] = self.user_feat[field_name].map(lambda x: map_dict.get(x, x))
             if self.item_feat is not None and field_name in self.item_feat.columns:
-                self.user_feat[field_name] = self.item_feat[field_name].map(lambda x: map_dict.get(x, x))
+                self.item_feat[field_name] = self.item_feat[field_name].map(lambda x: map_dict.get(x, x))
 
     def data_process_after_remap(self):
         """Data preprocessing, including: