Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

17 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Federated Learning Algorithms (Fashion-MNIST)

Python PyTorch License: MIT

Implementation of FedSGD-based Net, ResNet, LSTM, CNN, FCNN models and CNN-based FedSGD, FedAvg, SCAFFOLD, FedProx algorithms on the Fashion-MNIST dataset.


πŸ“‚ Project Structure

Distributed-Machine-Learning/
β”œβ”€β”€ algorithms/
β”‚   β”œβ”€β”€ FedSGD.ipynb
β”‚   β”œβ”€β”€ FedAvg.ipynb
β”‚   β”œβ”€β”€ FedProx.ipynb
β”‚   └── SCAFFOLD.ipynb
β”‚
β”œβ”€β”€ models/
β”‚   β”œβ”€β”€ Net.ipynb
β”‚   β”œβ”€β”€ ResNet.ipynb
β”‚   β”œβ”€β”€ LSTM.ipynb
β”‚   β”œβ”€β”€ CNN.ipynb
β”‚   β”œβ”€β”€ FCNN.ipynb
β”‚   └── KAN_CNN.ipynb
β”‚
β”œβ”€β”€ results/
β”‚   β”œβ”€β”€ Code experiment.pptx/
β”‚
└── README.md

About

Implement FedSGD-based Net, ResNet, LSTM, CNN, FCNN models and CNN-based FedSGD, FedAvg, SCAFFOLD, FedProx algorithms.

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages