AmirShabani commited on
Commit
36f3d6a
·
1 Parent(s): cd4c78c
Files changed (1) hide show
  1. core.py +1 -1
core.py CHANGED
@@ -158,7 +158,7 @@ def create_user_embedding(movie_ratings, movies_df):
158
  user_ratings_df['movieId'] = user_ratings_df.index
159
 
160
  # Merge the user_ratings_df with the movies_df to get the movie embeddings
161
- user_movie_embeddings = user_ratings_df.merge((movies_df).astype('float'), on='movieId', how='left')
162
 
163
  # Multiply the ratings with the movie embeddings
164
  user_movie_embeddings = user_movie_embeddings.iloc[:, 2:].values * user_movie_embeddings['rating'].values[:, np.newaxis]
 
158
  user_ratings_df['movieId'] = user_ratings_df.index
159
 
160
  # Merge the user_ratings_df with the movies_df to get the movie embeddings
161
+ user_movie_embeddings = (user_ratings_df).astype('float').merge(movies_df, on='movieId', how='left')
162
 
163
  # Multiply the ratings with the movie embeddings
164
  user_movie_embeddings = user_movie_embeddings.iloc[:, 2:].values * user_movie_embeddings['rating'].values[:, np.newaxis]