Skip to content
/ NN_impl Public

NN training framework implementation from scratch with backpropagation mechanism, Python.

Notifications You must be signed in to change notification settings

Yairz1/NN_impl

Repository files navigation

Deep Learning From Scratch

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.

Visualization of training process with our framework

צילום מסך 2020-10-14 122203

About

NN training framework implementation from scratch with backpropagation mechanism, Python.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages