From 89be22c39882eb332f8cfb2e70478852db1f47b3 Mon Sep 17 00:00:00 2001 From: Wicknight <61172796+Wicknight@users.noreply.github.com> Date: Wed, 8 Mar 2023 18:32:54 +0800 Subject: [PATCH] Update dataset.py --- recbole_cdr/data/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recbole_cdr/data/dataset.py b/recbole_cdr/data/dataset.py index 62f7a3f..31b03e2 100644 --- 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: