Skip to content

joel-foo/Neural_Network_From_Scratch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

Neural network from scratch

The only dependency required is NumPy.

  • Support for the following is provided
    • Forward pass and backprop
    • Batch and mini-batch gradient descent
    • Activation functions - sigmoid, relu, tanh, linear, softmax
    • Loss functions - MSE, binary cross entropy, cross entropy
  • Creating and training a model is simple (somewhat similar to the Tensorflow API):
    1. Create a list of Layer objects
    2. Pass the created list and necessary parameters to instantiate the NN object
    3. Call train on the NN object
  • Benchmarked using the MINST dataset
    • Simple network with a single hidden layer achieved 88% accuracy on the test set

About

Neural network from scratch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published