/Numpy-Neural-Network

A neural network implemented in pure numpy & python to work with MNIST dataset.

Primary LanguagePython

Overview

  • A neural network written in pure numpy to classify MNIST digits
  • Neural Network is customizable along with all the relevant hyperparameters under the config.yaml file
  • Run training with respect to the config file in config.yaml python main.py train
  • Run tests python main.py test

Features

  • Linear Layer
  • ReLU
  • Sigmoid
  • Softmax
  • Dropout
  • SGD
  • Adam