update app.py
Browse files
app.py
CHANGED
@@ -26,14 +26,13 @@ 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 =
|
30 |
-
|
31 |
|
32 |
# Limpiar la figura
|
33 |
plt.clf()
|
34 |
|
35 |
# Crear el gráfico de barras
|
36 |
-
plt.bar(genres, genre_counts
|
37 |
plt.xlabel('Género')
|
38 |
plt.ylabel('Cantidad')
|
39 |
plt.title('Cantidad de Películas por Género en las Recomendaciones')
|
@@ -51,9 +50,7 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
|
|
51 |
# Devolver la lista de títulos y el gráfico como una imagen
|
52 |
return ', '.join(recommended_titles), im
|
53 |
|
54 |
-
|
55 |
-
|
56 |
-
# Leer los datos
|
57 |
dfmerge = pd.read_csv('merged_data7.csv')
|
58 |
|
59 |
# Cargar el modelo
|
@@ -94,5 +91,4 @@ Te mostraremos algunas películas que pueden gustarte.</b></p>
|
|
94 |
)
|
95 |
|
96 |
# Lanzar la interfaz
|
97 |
-
demo.launch()
|
98 |
-
|
|
|
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()
|
33 |
|
34 |
# Crear el gráfico de barras
|
35 |
+
plt.bar(genres, genre_counts)
|
36 |
plt.xlabel('Género')
|
37 |
plt.ylabel('Cantidad')
|
38 |
plt.title('Cantidad de Películas por Género en las Recomendaciones')
|
|
|
50 |
# Devolver la lista de títulos y el gráfico como una imagen
|
51 |
return ', '.join(recommended_titles), im
|
52 |
|
53 |
+
# Leer los datos
|
|
|
|
|
54 |
dfmerge = pd.read_csv('merged_data7.csv')
|
55 |
|
56 |
# Cargar el modelo
|
|
|
91 |
)
|
92 |
|
93 |
# Lanzar la interfaz
|
94 |
+
demo.launch()
|
|