diff --git a/.gitignore b/.gitignore index 4ccaa027b330ceccd99b99eeda8dff2b001392bd..1e89a50e0ccba642512e901520c29ca1415a0142 100644 --- a/.gitignore +++ b/.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 diff --git a/README.md b/README.md index 1d1d3d24404a524e12eeeb34ae522e19f7a7caea..f447cbfc92b917cd80d615ae13313aa0482c37ef 100644 --- a/README.md +++ b/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 diff --git a/run_exp.py b/run_exp.py index cd988a0b96a4e1d8baecc02a81d0c7b18f88b7c7..b0399a699d489005483611f94c50929190c39708 100644 --- a/run_exp.py +++ b/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)