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 |