From 7175728568dedc06ed16a86affeeb5c58d953d16 Mon Sep 17 00:00:00 2001 From: Alec Helbling Date: Tue, 26 Apr 2022 12:11:38 -0400 Subject: [PATCH] Added a basic neural network example --- examples/basic_neural_network.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 examples/basic_neural_network.py diff --git a/examples/basic_neural_network.py b/examples/basic_neural_network.py new file mode 100644 index 0000000..d54df79 --- /dev/null +++ b/examples/basic_neural_network.py @@ -0,0 +1,22 @@ +from manim import * +from manim_ml.neural_network.layers import FeedForwardLayer +from manim_ml.neural_network.neural_network import NeuralNetwork + +class NeuralNetworkScene(Scene): + """Test Scene for the Neural Network""" + + def construct(self): + # Make the Layer object + layers = [ + FeedForwardLayer(3), + FeedForwardLayer(5), + FeedForwardLayer(3) + ] + nn = NeuralNetwork(layers) + nn.move_to(ORIGIN) + # Make Animation + self.add(nn) + #self.play(Create(nn)) + forward_propagation_animation = nn.make_forward_pass_animation(run_time=5, passing_flash=True) + + self.play(forward_propagation_animation) \ No newline at end of file