Spaces:
Running
Running
Update app.py
Browse files
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}\
|
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)
|