├── .gitignore ├── LICENSE ├── README.md ├── ascii_sequential_execution.py ├── core ├── __init__.py ├── evaluator │ ├── __init__.py │ └── model.py ├── federated_learning │ ├── __init__.py │ ├── aggregation │ │ ├── __init__.py │ │ ├── base.py │ │ ├── evaluatorv1.py │ │ ├── evaluatorv2.py │ │ └── fedavg.py │ ├── participation.py │ └── worker.py └── scheduler │ ├── __init__.py │ ├── actor_critic.py │ └── model.py ├── docs ├── example.png ├── report.pdf └── report.png ├── model4FL ├── __init__.py ├── cifar10.py ├── fashionmnist.py └── mnist.py ├── pyproject.toml ├── report.pdf ├── requirements.txt ├── tests ├── __init__.py ├── test_FL_classes.py ├── test_models.py ├── test_partition.py ├── test_path.py └── test_reinforce.py └── utils ├── __init__.py ├── distribution ├── __init__.py ├── common.py ├── iid.py ├── noniid.py └── partition.py ├── path.py └── plot.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/README.md -------------------------------------------------------------------------------- /ascii_sequential_execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/ascii_sequential_execution.py -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/__init__.py -------------------------------------------------------------------------------- /core/evaluator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/evaluator/__init__.py -------------------------------------------------------------------------------- /core/evaluator/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/evaluator/model.py -------------------------------------------------------------------------------- /core/federated_learning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/__init__.py -------------------------------------------------------------------------------- /core/federated_learning/aggregation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/aggregation/__init__.py -------------------------------------------------------------------------------- /core/federated_learning/aggregation/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/aggregation/base.py -------------------------------------------------------------------------------- /core/federated_learning/aggregation/evaluatorv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/aggregation/evaluatorv1.py -------------------------------------------------------------------------------- /core/federated_learning/aggregation/evaluatorv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/aggregation/evaluatorv2.py -------------------------------------------------------------------------------- /core/federated_learning/aggregation/fedavg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/aggregation/fedavg.py -------------------------------------------------------------------------------- /core/federated_learning/participation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/participation.py -------------------------------------------------------------------------------- /core/federated_learning/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/federated_learning/worker.py -------------------------------------------------------------------------------- /core/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/scheduler/__init__.py -------------------------------------------------------------------------------- /core/scheduler/actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/scheduler/actor_critic.py -------------------------------------------------------------------------------- /core/scheduler/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/core/scheduler/model.py -------------------------------------------------------------------------------- /docs/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/docs/example.png -------------------------------------------------------------------------------- /docs/report.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/docs/report.pdf -------------------------------------------------------------------------------- /docs/report.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/docs/report.png -------------------------------------------------------------------------------- /model4FL/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/model4FL/__init__.py -------------------------------------------------------------------------------- /model4FL/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/model4FL/cifar10.py -------------------------------------------------------------------------------- /model4FL/fashionmnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/model4FL/fashionmnist.py -------------------------------------------------------------------------------- /model4FL/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/model4FL/mnist.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/pyproject.toml -------------------------------------------------------------------------------- /report.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/report.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/requirements.txt -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/test_FL_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/tests/test_FL_classes.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/tests/test_partition.py -------------------------------------------------------------------------------- /tests/test_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/tests/test_path.py -------------------------------------------------------------------------------- /tests/test_reinforce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/tests/test_reinforce.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/distribution/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/distribution/__init__.py -------------------------------------------------------------------------------- /utils/distribution/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/distribution/common.py -------------------------------------------------------------------------------- /utils/distribution/iid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/distribution/iid.py -------------------------------------------------------------------------------- /utils/distribution/noniid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/distribution/noniid.py -------------------------------------------------------------------------------- /utils/distribution/partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/distribution/partition.py -------------------------------------------------------------------------------- /utils/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/path.py -------------------------------------------------------------------------------- /utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bourbonut/reinforcedFL/HEAD/utils/plot.py --------------------------------------------------------------------------------