Spaces:
Running
Running
vickeee465
commited on
Commit
·
079d9eb
1
Parent(s):
bba8601
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")
|