From 654f2781a3af75c468e0c3402037da605d741667 Mon Sep 17 00:00:00 2001 From: tqtg Date: Thu, 26 Oct 2023 22:17:10 +0000 Subject: [PATCH] Keep num_users and num_items, maybe changing to train_users and train_items in the future to be more explicit --- cornac/models/recommender.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cornac/models/recommender.py b/cornac/models/recommender.py index 301e9aab4..564e9eec9 100644 --- a/cornac/models/recommender.py +++ b/cornac/models/recommender.py @@ -42,10 +42,10 @@ class Recommender: Attributes ---------- - train_users: int + num_users: int Number of users in training data. - train_items: int + num_items: int Number of items in training data. total_users: int @@ -80,8 +80,8 @@ def __init__(self, name, trainable=True, verbose=False): self.ignored_attrs = [] # attributes to be ignored when saving model # useful information getting from train_set for prediction - self.train_users = None - self.train_items = None + self.num_users = None + self.num_items = None self.uid_map = None self.iid_map = None self.max_rating = None