AmirShabani
commited on
Commit
·
325a5ae
1
Parent(s):
8f0b5ca
Debug the input
Browse files
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
|
|
|
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)
|