Skip to content

Commit

Permalink
Keep num_users and num_items, maybe changing to train_users and train…
Browse files Browse the repository at this point in the history
…_items in the future to be more explicit
  • Loading branch information
tqtg committed Oct 26, 2023
1 parent 8a4f2f8 commit 654f278
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions cornac/models/recommender.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 654f278

Please sign in to comment.