From 6cf1f0b1eb185dfd961fc450fda1db7097d7db05 Mon Sep 17 00:00:00 2001
From: Leahcimali <michaelbenalipro@gmail.com>
Date: Tue, 29 Oct 2024 13:24:45 +0100
Subject: [PATCH] Update utils_results.py Added nn_model and number of clients
 to results summary

---
 src/utils_results.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/utils_results.py b/src/utils_results.py
index 5b08515..e724966 100644
--- a/src/utils_results.py
+++ b/src/utils_results.py
@@ -226,9 +226,9 @@ def summarize_results() -> None:
             
     df_results = pd.DataFrame(list_results)
     
-    df_results.sort_values(['dataset_type',  'dataset', 'exp_type', 'number_of_clients'], inplace=True)
+    df_results.sort_values(['dataset_type',  'dataset', 'exp_type', 'nn_model','number_of_clients'], inplace=True)
     
-    df_results = df_results[['exp_type', 'dataset', 'num_clusters', 'dataset_type', "accuracy", "ARI", "AMI", "hom", "cmplt", "vm"]]
+    df_results = df_results[['exp_type','nn_model','number_of_clients', 'dataset', 'num_clusters', 'dataset_type', "accuracy", "ARI", "AMI", "hom", "cmplt", "vm"]]
     
     df_results.to_csv("results/summarized_results.csv", float_format='%.2f', index=False, na_rep="n/a")
 
-- 
GitLab