File size: 795 Bytes
be9a8af c868436 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | ---
datasets:
- ylecun/mnist
- zalando-datasets/fashion_mnist
language:
- en
- ru
metrics:
- accuracy
- confusion_matrix
---
# How to run
Download mnist dataset: `python dataset_downloader.py -f "MNIST" -d "ylecun/mnist"`
Download fashion mnist dataset: `python dataset_downloader.py -f "Fashion-MNIST" -d "zalando-datasets/fashion_mnist"`
Train: `python mnist.py --mode train --dataset "./Fashion-MNIST/train.csv" --model "./models/Fashion-MNIST.pth" --epochs 8 --batch-size 64 --lr 0.001`
Inference: `python mnist.py --mode inference --input "./Fashion-MNIST/test.csv" --model "./models/Fashion-MNIST.pth" --output "./Fashion-MNIST/predictions.csv" --batch-size 64`
Evaluate: `python evaluate.py --ground-truth "./Fashion-MNIST/test.csv" --predictions "./Fashion-MNIST/predictions.csv"`
|