Merge pull request #5 from RUCAIBox/0.2.x

FIX: bug fix in avg_actions_of_users && avg_actions_of_items (RUCAIBox#576)
linzihan-backforward · Dec 12, 2020 · 0024efb · 0024efb
2 parents 7518311 + ac4a9f6
commit 0024efb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recbole/data/dataset/dataset.py b/recbole/data/dataset/dataset.py
@@ -1108,7 +1108,7 @@ def avg_actions_of_users(self):
         Returns:
             numpy.float64: Average number of users' interaction records.
         """
-        return np.mean(list(Counter(self.inter_feat[self.uid_field]).values()))
+        return np.mean(list(Counter(self.inter_feat[self.uid_field].numpy()).values()))
 
     @property
     def avg_actions_of_items(self):
@@ -1117,7 +1117,7 @@ def avg_actions_of_items(self):
         Returns:
             numpy.float64: Average number of items' interaction records.
         """
-        return np.mean(list(Counter(self.inter_feat[self.iid_field]).values()))
+        return np.mean(list(Counter(self.inter_feat[self.iid_field].numpy()).values()))
 
     @property
     def sparsity(self):