wi-lab commited on
Commit
25155a2
·
verified ·
1 Parent(s): f3ce7b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -92,10 +92,9 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme='Dark'):
92
 
93
  cbar = ax.collections[0].colorbar
94
  cbar.ax.yaxis.set_tick_params(color=text_color)
95
- plt.setp(cbar.ax.yaxis.get_ticklabels(), color=text_color)
96
 
97
  # Add F1-score to the title
98
- plt.title(f"{title}\nF1 Score: {avg_f1:.3f}", color=text_color, fontsize=14)
99
 
100
  tick_marks = np.arange(len(classes))
101
  plt.xticks(tick_marks, classes, color=text_color, fontsize=14) # Adjust text color based on the mode
 
92
 
93
  cbar = ax.collections[0].colorbar
94
  cbar.ax.yaxis.set_tick_params(color=text_color)
 
95
 
96
  # Add F1-score to the title
97
+ plt.title(f"{title}\nF1 Score: {avg_f1:.3f}", color=text_color, fontsize=23)
98
 
99
  tick_marks = np.arange(len(classes))
100
  plt.xticks(tick_marks, classes, color=text_color, fontsize=14) # Adjust text color based on the mode