Module nn

Source
Expand description

A simple implementation of simple neural network’s elements.

This package includes the following elements to construct neural networks:

  • Neuron: weight and bias
  • Layer: a collection of neurons
  • MLP: multilayer-perceptron, a collection of layers and activation function

All of them have a forward method to calculate the output of the element.

Structs§

Layer
A layer in a neural network.
MLP
A multilayer perceptron.
Neuron
A neuron in a neural network.

Enums§

Activation
Activation functions for neurons.