AmirShabani commited on
Commit
b487903
·
1 Parent(s): 36f3d6a

Update core.py

Browse files
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).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]
 
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]