Neural Network Builder

Build and train a standard fully-connected neural network. Choose the size of input, number of layers, type of activation function, and number of output classes in your network Train your network by providing data inputs and inputs with a specific learning rate. Use your fully trained network to perform classification or regression!

Getting Started

Prerequisites

Import NumPy

Usage

Creating architecture

Create a NeuralNetwork object by specifying the number of nodes in each layer as an array (e.g. [256, 60, 10] for an array with input size of 256, hidden layer size of 60 and output size of 10). Specify the learning rate you want to use at when creating the NeuralNetwork.

Training

Call the train function on the neural network using a set of data. The number of inputs and outputs should match. Specify the number of epochs to be used in training.

Saving and Reading NN object

Functionality coming soon...