Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme='Dark'):
|
|
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=
|
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
|
|
|
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
|