DavidHosp commited on
Commit
2f26661
·
1 Parent(s): 2b8a291

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -28,11 +28,13 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
28
  recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
29
  recommended_genres = [df[df['id'] == movie_id]['genre'].values[0] for movie_id, _ in recommended_movies[:top]]
30
 
31
- # Crear un dataframe con las recomendaciones
32
  recommendations_df = pd.DataFrame({
33
  'Top': range(1, top+1),
34
  'Películas': recommended_titles,
35
  'Genero': recommended_genres
 
 
36
 
37
  # Contar cuántas películas de cada género hay en las recomendaciones
38
  recommended_movies_ids = [movie_id for movie_id, _ in recommended_movies[:top]]
 
28
  recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
29
  recommended_genres = [df[df['id'] == movie_id]['genre'].values[0] for movie_id, _ in recommended_movies[:top]]
30
 
31
+ # Crear un dataframe con las recomendaciones
32
  recommendations_df = pd.DataFrame({
33
  'Top': range(1, top+1),
34
  'Películas': recommended_titles,
35
  'Genero': recommended_genres
36
+ })
37
+
38
 
39
  # Contar cuántas películas de cada género hay en las recomendaciones
40
  recommended_movies_ids = [movie_id for movie_id, _ in recommended_movies[:top]]