Skip to content

This project looks over the basic structure of a neural network and how machine learning training algorithms work. For this project, I used the sigmoid function as an activation function along with stochastic gradient descent to adjust the weights and biases.

Notifications You must be signed in to change notification settings

ManasBommakanti/BasicNeuralNetwork

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BasicNeuralNetwork

This project looks over the basic structure of a neural network and how machine learning training algorithms work. For this project, I used the sigmoid function as an activation function along with stochastic gradient descent to adjust the weights and biases.

This neural network predicts the gender of an individual (either male or female) based on their height and weight.

Training Time and Loss

Time to train neural network: Around 1 to 2 sec

Graph shows Mean-Squared-Error (MSE) loss

Graph shows Mean-Squared-Error (MSE) loss over time. As the loss decreases over time, the network has a better understanding of how to classify the gender of an individual.

References

3Blue1Brown's videos on Neural Networks

Victor Zhou's Introduction to Neural Networks

About

This project looks over the basic structure of a neural network and how machine learning training algorithms work. For this project, I used the sigmoid function as an activation function along with stochastic gradient descent to adjust the weights and biases.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages