Skip to content

REGATTE/Federated-Learning-with-TensorFlow

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is in part to the medium blog

Instructions

Set up the environment with requirements.txt(pip) or environment.yml(conda)

Use npm or yarn to install node dependencies.

Download dataset and extract:

http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz

http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz

Run in following order.

python create_model.py
./to_tfjs.sh
node train_on_cpus.js
node convert_tfjs_to_keras.js
python client_fed_average.py

youtube video link: https://youtu.be/iihM0-rcXrM