opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-05-17 18:55:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added support for Neural Network overhead title and per-layer title.

Browse Source
This commit is contained in:
Alec Helbling
2022-04-19 00:36:15 -04:00
parent 05f512f185
commit 229c27fa3f
12 changed files with 77 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_layers.py Normal file
View File

@ -0,0 +1,12 @@
from manim import *
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
from manim_ml.neural_network.layers.feed_forward_to_feed_forward import FeedForwardToFeedForward
from manim_ml.neural_network.layers.util import get_connective_layer
def test_get_connective_layer():
"""Tests get connective layer"""
input_layer = FeedForwardLayer(3)
output_layer = FeedForwardLayer(5)
connective_layer = get_connective_layer(input_layer, output_layer)
assert isinstance(connective_layer, FeedForwardToFeedForward)

3
tests/test_neural_network.py
View File

@ -94,7 +94,7 @@ class NeuralNetworkScene(Scene):
def construct(self):
# Make the Layer object
layers = [
FeedForwardLayer(3),
FeedForwardLayer(3, title="Title Test"),
FeedForwardLayer(5),
FeedForwardLayer(3)
]
@ -102,6 +102,7 @@ class NeuralNetworkScene(Scene):
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)