Commit
·
b487903
1
Parent(s):
36f3d6a
Update core.py
Browse files
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).
|
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).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]
|