Working neural network test with refactor

This commit is contained in:
Alec Helbling
2022-04-02 19:20:30 -04:00
parent 861ae77002
commit 61b47798f3
4 changed files with 63 additions and 12 deletions

View File

@ -0,0 +1,39 @@
from manim import *
from manim_ml.neural_network.layers import FeedForwardLayer
from manim_ml.neural_network.neural_network import NeuralNetwork
from manim_ml.neural_network.feed_forward import FeedForwardNeuralNetwork
config.pixel_height = 720
config.pixel_width = 1280
config.frame_height = 6.0
config.frame_width = 6.0
class FeedForwardNeuralNetworkScene(Scene):
def construct(self):
nn = FeedForwardNeuralNetwork([3, 5, 3])
self.play(Create(nn))
self.play(Wait(3))
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)
forward_propagation_animation = nn.make_forward_pass_animation(run_time=5, passing_flash=True)
self.play(forward_propagation_animation)
if __name__ == "__main__":
"""Render all scenes"""
# Feed Forward Neural Network
ffnn_scene = FeedForwardNeuralNetworkScene()
ffnn_scene.render()
# Neural Network
nn_scene = NeuralNetworkScene()
nn_scene.render()