Skip to content

Neural network that classify images from CIFAR10 dataset into 10 different classes. Technologies: Python, pyTorch, scikit-learn, numPy.

Notifications You must be signed in to change notification settings

AdamJedz/NeuralNet_Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural Net image classifier

Description

Neural network to classify images from CIFAR10 dataset into 10 different classes.

Instruction

Trained network is in model directory (test_best.py). To test it on test data run get_predictions.py.
To train new network run model.py. You can perform changes in network in network.py file.

Current accuracy on test set is around 83%.

About

Neural network that classify images from CIFAR10 dataset into 10 different classes. Technologies: Python, pyTorch, scikit-learn, numPy.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published