main module
Script to run the baselines.
-
main.main()[source]
-
main.online(clients)[source]
We assume all users are always online.
-
main.parse_args()[source]
-
main.print_metrics(metrics, weights)[source]
Prints weighted averages of the given metrics.
| Parameters: |
- metrics – dict with client ids as keys. Each entry is a dict
with the metrics of that client.
- weights – dict with client ids as keys. Each entry is the weight
for that client.
|
-
main.save_model(server_model, dataset, model)[source]
Saves the given server model on checkpoints/dataset/model.ckpt.
-
main.setup_clients(dataset, model=None)[source]
Instantiates clients based on given train and test data directories.
| Returns: | list of Client objects. |
| Return type: | all_clients |