AmirShabani commited on
Commit
c3dd99d
·
1 Parent(s): 42428e3

Test user embeddings

Browse files
Files changed (1) hide show
  1. core.py +2 -1
core.py CHANGED
@@ -159,7 +159,8 @@ def create_user_embedding(movie_ratings, movies_df):
159
  # Convert the movie_ratings dictionary to a dataframe
160
  user_ratings_df = pd.DataFrame.from_dict(movie_ratings, orient='index', columns=['rating'])
161
  user_ratings_df['movieId'] = user_ratings_df.index
162
-
 
163
  # Merge the user_ratings_df with the movies_df to get the movie embeddings
164
  user_movie_embeddings = user_ratings_df.merge(movies_df, on='movieId', how='left')
165
 
 
159
  # Convert the movie_ratings dictionary to a dataframe
160
  user_ratings_df = pd.DataFrame.from_dict(movie_ratings, orient='index', columns=['rating'])
161
  user_ratings_df['movieId'] = user_ratings_df.index
162
+ print(user_ratings_df)
163
+ print(user_movie_embeddings)
164
  # Merge the user_ratings_df with the movies_df to get the movie embeddings
165
  user_movie_embeddings = user_ratings_df.merge(movies_df, on='movieId', how='left')
166