Skip to content

gosuhiman/mnist-classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mnist-classifier

Model

model = Sequential([
    Conv2D(32, (5, 5), activation='relu', input_shape=(28, 28, 1)),
    MaxPooling2D(pool_size=(3, 3)),
    Conv2D(32, (3, 3), activation='relu'),
    MaxPooling2D(pool_size=(2, 2)),
    Flatten(),
    Dense(48, activation='relu'),
    Dense(10, activation='softmax'),
])

MNIST test data set - 0.9866 accuracy

MNIST data set confusion matrix

My own digits data set - 0.9615 accuracy

My own data set confusion matrix

No augmentation

Without augmentation (train-pure.py) I get higher accuracy 0.99++ on mnist dataset, but it fails miserably on my own data set.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages