Skip to content

Sequential neural network with PyTorch for image classification of FMNIST dataset, trained and validated over 30 epochs using 1200 image batch size.

Notifications You must be signed in to change notification settings

evanshlom/PyTorch-NN-Classification-Validation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

PyTorch NN Classification Validation

Sequential Neural Network using PyTorch

Image Dataset

The Fashion-MNIST dataset contains 60,000 training images (and 10,000 test images) of fashion and clothing items, taken from 10 classes. Each image is a standardized 28x28 size in grayscale (784 total pixels).

Virtual Environment Setup

To run the program, copy the virtual environment in this repository's yml file: using the Anaconda command line, enter the command below.

conda env create --name torch_copy_env -f torch_env.yml

Model Accuracy

After 30 epochs, the neural network's training accuracy was 94% and validation accuracy was 88%.

About

Sequential neural network with PyTorch for image classification of FMNIST dataset, trained and validated over 30 epochs using 1200 image batch size.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages