Simplified namespace so layers can be imported from manim_ml.neural_network

This commit is contained in:
Alec Helbling
2023-01-27 14:20:11 -05:00
parent 46958ea293
commit b767b9abee
8 changed files with 54 additions and 21 deletions

View File

@ -0,0 +1,14 @@
from manim import *
from manim_ml.neural_network import NeuralNetwork, FeedForwardLayer
class FeedForwardScene(Scene):
def construct(self):
nn = NeuralNetwork([
FeedForwardLayer(3),
FeedForwardLayer(5),
FeedForwardLayer(3)
])
self.add(nn)