Skip to content

Very simple neural nets, two and three layers, using only NumPy

Notifications You must be signed in to change notification settings

waynemystir/simple-neural-nets

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

simple-neural-nets

This repo is derived from the blog by @iamtrask.

I added and changed a few things for educational purposes. In particular, I spelled out key functions, modified the update rule, and plotted out the losses and weights. I also added "rev" scripts which flip the matrices and dot products.

As @iamtrask mentioned, I found it very helpful to try writing these networks from memory.

Usage

Each script has an argument for plotting:

python two_layer_nn.py --plot

python three_layer_nn.py --plot

About

Very simple neural nets, two and three layers, using only NumPy

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages