kaggle digit-recognizer CNN implementation

Environment

python 3.9.12 Aug 20, 2023

Usage

make sure your structure is as follows:

digit-recognizer.
├─data
│  ├─test.csv
│  └─train.csv
├─model
│  
├─output
│  
└─main.py

• dataset from Kaggle Digit Recognizer challenge is stored in data directory.

Now run the main.py Then

digit-recognizer.
├─data
│  ├─test.csv
│  └─train.csv
├─model
│  └─CNN.pt
├─output
│  └─submission.csv
└─main.py

• The trianed model is stored in model directory.
• The prediction is stored in output directory.

hyper parameters

loss function: CrossEntropyLoss

optimizer: Adam

learnig rate: 0.003

batch size: 32

epochs:20

dropout:0.25

You can ajust the hyper parameters or choose other loss functions and optimizers.