mirror of
https://github.com/helblazer811/ManimML.git
synced 2025-05-17 18:55:54 +08:00
Working neural network test with refactor
This commit is contained in:
39
tests/test_neural_network.py
Normal file
39
tests/test_neural_network.py
Normal 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()
|
Reference in New Issue
Block a user