DavidHosp commited on
Commit
cf139ac
·
1 Parent(s): fc26408

update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -26,7 +26,10 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
26
  recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
27
 
28
  # Contar cuántas películas de cada género hay en las recomendaciones
29
- genre_counts = df_filtered[df_filtered['id'].isin([movie_id for movie_id, _ in recommended_movies])][genres].sum()
 
 
 
30
 
31
  # Limpiar la figura
32
  plt.clf()
 
26
  recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
27
 
28
  # Contar cuántas películas de cada género hay en las recomendaciones
29
+ # genre_counts = df_filtered[df_filtered['id'].isin([movie_id for movie_id, _ in recommended_movies])][genres].sum()
30
+
31
+ recommended_movies_ids = [movie_id for movie_id, _ in recommended_movies[:top]]
32
+ genre_counts = df_filtered[df_filtered['id'].isin(recommended_movies_ids)][genres].sum()
33
 
34
  # Limpiar la figura
35
  plt.clf()