Download and place the dataset in files folder of project root.
Folder structure -
- mnist
- files
- dataset
- test.csv
- train.csv
- ...
Location: https://www.kaggle.com/c/digit-recognizer/data
On local machine
pip3 install -r requirements.txt
With docker-compose
docker-compose up --build -d
python main.py --mode train
python main.py --mode test