/cifar10-classification

Small Tensorflow model for CIFAR-10 dataset classification

Primary LanguagePython

CIFAR-10 Classification

Install

Enter virtual environment of your liking and install the requirements:

pip install -r requirements.txt

Run the model training script:

python main.py

Results

Training History

Training History

Confusion Matrix

Confusion Matrix