Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ df_metrics = df.pivot_table(index='dataset', columns='model', values=['accuracy'
|
|
| 8 |
|
| 9 |
# Reorganize columns
|
| 10 |
cols_order = []
|
| 11 |
-
metrics = ['accuracy', 'precision', 'recall', '
|
| 12 |
models = ['DecisionTree', 'RandomForest', 'SVM', 'KNN', 'LogisticRegression']
|
| 13 |
|
| 14 |
for metric in metrics:
|
|
@@ -19,11 +19,16 @@ df_metrics = df_metrics[cols_order]
|
|
| 19 |
|
| 20 |
# Rename columns for better readability
|
| 21 |
new_cols = []
|
| 22 |
-
for
|
| 23 |
-
|
| 24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
|
| 26 |
df_metrics.columns = new_cols
|
|
|
|
| 27 |
df_metrics = df_metrics.sort_index(key=lambda x: x.str[1:].astype(int))
|
| 28 |
df_compare = pd.read_csv("pairwise_comparison_results.csv")
|
| 29 |
|
|
|
|
| 8 |
|
| 9 |
# Reorganize columns
|
| 10 |
cols_order = []
|
| 11 |
+
metrics = ['accuracy', 'precision', 'recall', 'f1_score']
|
| 12 |
models = ['DecisionTree', 'RandomForest', 'SVM', 'KNN', 'LogisticRegression']
|
| 13 |
|
| 14 |
for metric in metrics:
|
|
|
|
| 19 |
|
| 20 |
# Rename columns for better readability
|
| 21 |
new_cols = []
|
| 22 |
+
for metric, model in df_metrics.columns:
|
| 23 |
+
model_short = model.replace("DecisionTree", "DT") \
|
| 24 |
+
.replace("RandomForest", "RF") \
|
| 25 |
+
.replace("LogisticRegression", "LR")
|
| 26 |
+
|
| 27 |
+
metric_name = "F1 Score" if metric == "f1_score" else metric.capitalize()
|
| 28 |
+
new_cols.append(f"{model_short} {metric_name}")
|
| 29 |
|
| 30 |
df_metrics.columns = new_cols
|
| 31 |
+
|
| 32 |
df_metrics = df_metrics.sort_index(key=lambda x: x.str[1:].astype(int))
|
| 33 |
df_compare = pd.read_csv("pairwise_comparison_results.csv")
|
| 34 |
|