Generates a static diagram of a neural network, where each neuron is connected to every neuron in the previous layer.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore
LICENSE.md
README.md
main.py

README.md

Visualise Neural Network

Generates a static diagram of a neural network, where each neuron is connected to every neuron in the previous layer.

For example, this code:

network = NeuralNetwork()
network.add_layer(3)
network.add_layer(4)
network.add_layer(1)
network.draw()

Will generate this diagram:

Diagram of a neural network with 3 neurons in the first layer, 4 neurons in the second layer and 1 neuron in the 3rd layer