Files
ManimML/tests/test_feed_forward.py
2023-01-31 10:27:51 -05:00

17 lines
441 B
Python

from manim import *
from manim_ml.utils.testing.frames_comparison import frames_comparison
from manim_ml.neural_network import NeuralNetwork, FeedForwardLayer
__module_test__ = "feed_forward"
@frames_comparison
def test_FeedForwardScene(scene):
"""Tests the appearance of a feed forward network"""
nn = NeuralNetwork([
FeedForwardLayer(3),
FeedForwardLayer(5),
FeedForwardLayer(3)
])
scene.add(nn)