This project includes implementation of a simple neural network for classification tasks and backpropagation mechanism.
The main building blocks are the following modules:
• Tests folder - includes gradient based tests for backpropagation mechanism.
• Neural Network.py - standard neural network.
• function.py - includes all activation functions and its gradients.
• loss function.py - objective loss function "soft-max" and its gradient.
• optimizer.py - implementation of stochastic gradient decent optimizer.
• train.py - implementation of training functions.
• Utils.py - util functions.
• main.py - full network experiments implementation.
-
Notifications
You must be signed in to change notification settings - Fork 0
Yairz1/NN_impl
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
NN training framework implementation from scratch with backpropagation mechanism, Python.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published