Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
1fc074a
1
Parent(s):
7b8cad9
fixing heatmap axes
Browse files
app.py
CHANGED
|
@@ -133,7 +133,9 @@ def plot_emotion_heatmap(heatmap_data):
|
|
| 133 |
)
|
| 134 |
|
| 135 |
# Format axis
|
|
|
|
| 136 |
ax.set_xticklabels(heatmap_data.columns, rotation=0, ha='center')
|
|
|
|
| 137 |
ax.set_yticklabels(heatmap_data.index, rotation=0, ha='right')
|
| 138 |
|
| 139 |
ax.set_xlabel("Sentences")
|
|
|
|
| 133 |
)
|
| 134 |
|
| 135 |
# Format axis
|
| 136 |
+
ax.set_xticks(np.arange(len(heatmap_data.columns)) + 0.5)
|
| 137 |
ax.set_xticklabels(heatmap_data.columns, rotation=0, ha='center')
|
| 138 |
+
ax.set_yticks(np.arange(len(heatmap_data.index)) + 0.5)
|
| 139 |
ax.set_yticklabels(heatmap_data.index, rotation=0, ha='right')
|
| 140 |
|
| 141 |
ax.set_xlabel("Sentences")
|