vickeee465 commited on
Commit
0c4f21d
·
1 Parent(s): f382753

heatmap colormap (viridis)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def prepare_heatmap_data(data):
91
 
92
  def plot_emotion_heatmap(heatmap_data):
93
  fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 4, 5))
94
- sns.heatmap(heatmap_data, annot=False, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
95
  plt.xlabel("Sentences")
96
  plt.ylabel("Emotions")
97
  plt.xticks(rotation=45)
 
91
 
92
  def plot_emotion_heatmap(heatmap_data):
93
  fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 4, 5))
94
+ sns.heatmap(heatmap_data, annot=False, cmap="viridis", cbar=True, linewidths=0.5, linecolor='gray')
95
  plt.xlabel("Sentences")
96
  plt.ylabel("Emotions")
97
  plt.xticks(rotation=45)