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

@ -1,8 +1,6 @@
from manim import *
from manim_ml.neural_network.layers.convolutional_2d import Convolutional2DLayer
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
from manim_ml.neural_network.neural_network import NeuralNetwork
from manim_ml.neural_network import Convolutional2DLayer, FeedForwardLayer, NeuralNetwork
# Make the specific scene
config.pixel_height = 700