arbdigital / visualise-neural-network

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

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: