This is in part to the medium blog
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
--------------- TEAM ---------------
| Name | Enrollment id |
|---|---|
| Jai Chhabra | E18CSE071 |
| Aniket Biprojit | E18CSE015 |
| Aaryan Kapur | E18CSE004 |
| Abhinav Robinson | E18CSE006 |
| Ashok Kumar | E18CSE029 |