Skip to content

Model library for tracebloc's collaboration training workflow

Notifications You must be signed in to change notification settings

tracebloc/model-zoo

Repository files navigation

#Jupyter notebook for Data Scientist Data scientist tool to create and run experiment with required model and training plan .

Pre-requisite

###1. Install Ananconda

wget https://repo.anaconda.com/archive/Anaconda3-2020.11-Linux-x86_64.sh 

bash Anaconda3-2020.11-Linux-x86_64.sh -b -p -y 

source ~/.bashrc 

###2. Create an environment and activate it

conda create -n dsenvironemnt python=3.8 

conda activate dsenvironemnt 

###3. Install required package

pip install -r requirement.txt

###4. Install tensorflow and keras

pip install tensorflow==2.6.0
pip install keras==2.6.0

####In case of mac m1 machine

Follow this link to install tensorflow: https://caffeinedev.medium.com/how-to-install-tensorflow-on-m1-mac-8e9b91d93706

or

Enter following commands

conda install -c apple tensorflow-deps
pip install tensorflow-macos
pip install tensorflow-metal
pip uninstall -y numpy
pip uninstall -y setuptools
pip install numpy
pip install setuptools

###4. Install Jupyter Notebook

conda install jupyter notebook

pip install jupyter

###5. Move to Directory

cd federated-xray-datascientist

###6. Start Jupyter Notebook

jupyter notebook

How to run an experiment

Follow this document guide to learn how to create and run an experiment.

Testing Jupyter Notebook

In order to start and test experiment, a model file required.

For this purpose one model is placed in directory ("CNN.py")

There are some other test models in this repo in test-models directory.

In order to test those copy the model file from test-models directory to parent directory.