Skip to content

A neural network to predict the category of the item in an image. Trained using the Fashion-MNIST dataset.

Notifications You must be signed in to change notification settings

Invictus17/NeuralNetworkFromScratch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 

Repository files navigation

Implemented a Neural Netwrok from scratch with RMSprop

Task: Identify the class of the object in the image out of 10 possible classes.

  • Network architecture:
    Input layer
    2 Dense hidden layers with 512 neurons, using relu as the activation function
    Dropout with a value of 0.2
    Output layer, using softmax as the activation function

Achieved testing accuarcy of 84 % on Fashion MNIST dataset

About

A neural network to predict the category of the item in an image. Trained using the Fashion-MNIST dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages