MorettiGS commited on
Commit
3e8662d
1 Parent(s): 61af90b

fix format

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,10 +12,10 @@ ids_list = list(map(str, ids.tolist()))
12
  ratings = pd.read_csv('ratings.csv')
13
 
14
  def top5(user):
 
15
  items = pd.Series(learn.dls.classes['title']).unique()
16
 
17
  clas_items = ratings.loc[(ratings['user'] == user) & (ratings['rating'] > 0), 'title']
18
-
19
  no_clas_items = np.setdiff1d(items, clas_items)
20
 
21
  df = pd.DataFrame({'user': [user]*len(no_clas_items), 'title': no_clas_items})
 
12
  ratings = pd.read_csv('ratings.csv')
13
 
14
  def top5(user):
15
+ user = int(user)
16
  items = pd.Series(learn.dls.classes['title']).unique()
17
 
18
  clas_items = ratings.loc[(ratings['user'] == user) & (ratings['rating'] > 0), 'title']
 
19
  no_clas_items = np.setdiff1d(items, clas_items)
20
 
21
  df = pd.DataFrame({'user': [user]*len(no_clas_items), 'title': no_clas_items})