Merge branch 'main' into Occidata

.gitignore

@@ -12,5 +12,8 @@ pub/*
 data/*
 *.tgz
 *.pyc
-
-src/__pycache__/*
\ No newline at end of file
+info.log
+*.ipynb
+launch.json
+backup_results/*
+*.sh
\ No newline at end of file

README.md

@@ -8,5 +8,10 @@
 
 3. Histogram plots and a summary table of various experiments can be obtained running `src/utils_results.py`
   
+To use driver.py use the following parameters : 
 
-  
+`python driver.py --exp_type --dataset --heterogeneity_type  --num_clients --num_samples_by_label --num_clusters --centralized_epochs --federated_rounds --seed ` 
+
+To run all experiments in exp_config.csv user `run_exp.py`. 
+
+Once all experiments are done, to get results run `src/utils_results.src`.
\ No newline at end of file

run_exp.py

@@ -20,6 +20,7 @@ with open(csv_file, newline='') as csvfile:
         "python", "driver.py",
         "--exp_type", exp_type,
         "--dataset", dataset,
+
         "--nn_model", nn_model,
         "--heterogeneity_type", heterogeneity_type,
         "--num_clients", num_clients,
@@ -27,8 +28,7 @@ with open(csv_file, newline='') as csvfile:
         "--num_clusters", num_clusters,
         "--centralized_epochs", centralized_epochs,
         "--federated_rounds", federated_rounds,
-        "--seed", seed
-    ]
+        "--seed", seed]
 
     # Run the command
     subprocess.run(command)