DavidHosp commited on
Commit
1a93be5
·
1 Parent(s): 8188aee

update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
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()