tobiasaurer commited on
Commit
ea43f50
1 Parent(s): 686aaf3

minor bugfix

Browse files
pages/3 - Similarity-Based Recommender.py CHANGED
@@ -37,7 +37,9 @@ Note: The more genres you choose, the fewer movies will be recommended.
37
  def get_similar_recommendations(movie_title, n, genres):
38
 
39
  # select similarity for chosen movie
40
- similarities = pd.DataFrame(similarities_movies.loc[similarities_movies.index != movie_title, movie_title])
 
 
41
 
42
  # exclude genres if necessary and return the n movies with the highest similarity
43
  recommendations = (
 
37
  def get_similar_recommendations(movie_title, n, genres):
38
 
39
  # select similarity for chosen movie
40
+ similarities = pd.DataFrame(
41
+ (similarities_movies.query("index != @movie_title")[movie_title] / sum(similarities_movies.query("index != @movie_title")[movie_title]))
42
+ .sort_values(ascending= False))
43
 
44
  # exclude genres if necessary and return the n movies with the highest similarity
45
  recommendations = (