wi-lab commited on
Commit
3c003bc
·
verified ·
1 Parent(s): 7c0043a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -191,7 +191,7 @@ def plot_confusion_matrix_from_csv(csv_file_path, title, save_path, light_mode=F
191
  sns.heatmap(cm, annot=True, fmt="d", cmap=cmap, cbar=False, annot_kws={"size": 12}, linewidths=0.5, linecolor='white')
192
 
193
  # Add F1-score to the title
194
- plt.title(f"{title}\n(F1 Score: {f1:.3f})", color=text_color, fontsize=18)
195
 
196
  # Customize tick labels for light/dark mode
197
  plt.xticks([0.5, 1.5], labels=['Class 0', 'Class 1'], color=text_color, fontsize=12)
 
191
  sns.heatmap(cm, annot=True, fmt="d", cmap=cmap, cbar=False, annot_kws={"size": 12}, linewidths=0.5, linecolor='white')
192
 
193
  # Add F1-score to the title
194
+ plt.title(f"{title}\nF1 Score: {f1:.3f}", color=text_color, fontsize=18)
195
 
196
  # Customize tick labels for light/dark mode
197
  plt.xticks([0.5, 1.5], labels=['Class 0', 'Class 1'], color=text_color, fontsize=12)