vickeee465 commited on
Commit
079d9eb
·
1 Parent(s): bba8601

fixing heatmap axes

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