AmirShabani commited on
Commit
325a5ae
·
1 Parent(s): 8f0b5ca

Debug the input

Browse files
Files changed (1) hide show
  1. core.py +2 -1
core.py CHANGED
@@ -185,7 +185,8 @@ def drop_non_numerical_columns(df):
185
 
186
  def output_list(input_dict, movies_df = movie_embeds, tree = btree, user_embeddings = user_embeds, movies = final_movies):
187
  movie_ratings = {}
188
- for movie_title, rating in input_dict:
 
189
  index = movies.index[movies['title'] == movie_title].tolist()[0]
190
  movie_ratings[index] = rating
191
  user_embed = create_user_embedding(movie_ratings, movie_embeds)
 
185
 
186
  def output_list(input_dict, movies_df = movie_embeds, tree = btree, user_embeddings = user_embeds, movies = final_movies):
187
  movie_ratings = {}
188
+ for movie_title in input_dict:
189
+ print(str(movie_title))
190
  index = movies.index[movies['title'] == movie_title].tolist()[0]
191
  movie_ratings[index] = rating
192
  user_embed = create_user_embedding(movie_ratings, movie_embeds)