Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
c5a271e
1
Parent(s):
6efba5d
heatmap colormap (viridis)
Browse files
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="
|
| 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)
|