Navinor commited on
Commit
0d518f7
·
verified ·
1 Parent(s): 9de310b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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', 'F1_score']
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 col in df_metrics.columns:
23
- metric, model = col
24
- new_cols.append(f'{model.replace("DecisionTree", "DT").replace("RandomForest", "RF").replace("LogisticRegression", "LR")} {metric.capitalize()}')
 
 
 
 
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