update app.py
Browse files
app.py
CHANGED
@@ -36,11 +36,19 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
|
|
36 |
genre_colors = {'Drama': 'blue', 'Comedy': 'orange', 'Horror': 'red', 'Romance': 'pink'}
|
37 |
colors = [genre_colors[genre] for genre in genres]
|
38 |
|
|
|
|
|
39 |
# Crear el gráfico de barras con los colores específicos
|
|
|
40 |
plt.bar(genres, genre_counts, color=colors)
|
41 |
-
plt.xlabel('Género')
|
42 |
-
plt.ylabel('Cantidad')
|
43 |
-
plt.title('Cantidad de Películas por Género en las Recomendaciones')
|
|
|
|
|
|
|
|
|
|
|
44 |
|
45 |
# Guardar el gráfico como una imagen PNG en una cadena de bytes
|
46 |
buf = io.BytesIO()
|
|
|
36 |
genre_colors = {'Drama': 'blue', 'Comedy': 'orange', 'Horror': 'red', 'Romance': 'pink'}
|
37 |
colors = [genre_colors[genre] for genre in genres]
|
38 |
|
39 |
+
|
40 |
+
|
41 |
# Crear el gráfico de barras con los colores específicos
|
42 |
+
plt.style.use('ggplot') # establece el estilo del gráfico
|
43 |
plt.bar(genres, genre_counts, color=colors)
|
44 |
+
plt.xlabel('Género', fontsize=14)
|
45 |
+
plt.ylabel('Cantidad', fontsize=14)
|
46 |
+
plt.title('Cantidad de Películas por Género en las Recomendaciones', fontsize=16)
|
47 |
+
plt.grid(True) # agrega una cuadrícula
|
48 |
+
plt.xticks(fontsize=12) # ajusta el tamaño de la fuente de los ticks del eje x
|
49 |
+
plt.yticks(fontsize=12) # ajusta el tamaño de la fuente de los ticks del eje y
|
50 |
+
|
51 |
+
|
52 |
|
53 |
# Guardar el gráfico como una imagen PNG en una cadena de bytes
|
54 |
buf = io.BytesIO()
|