opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-06-05 19:15:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactored Neural Network Layers into their own files.

Browse Source
This commit is contained in:
Alec Helbling
2022-04-14 01:53:12 -04:00
parent b1490c0117
commit 7be4dfc8a4
18 changed files with 445 additions and 464 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_variational_autoencoder.py
View File

@ -1,6 +1,6 @@
from manim import *
from PIL import Image
from manim_ml.neural_network.embedding import EmbeddingLayer, GaussianDistribution
from manim_ml.neural_network.embedding import EmbeddingLayer
from manim_ml.neural_network.feed_forward import FeedForwardLayer
from manim_ml.neural_network.image import ImageLayer
from manim_ml.neural_network.neural_network import NeuralNetwork
@ -10,7 +10,7 @@ config.pixel_width = 1280
config.frame_height = 6.0
config.frame_width = 6.0
class GaussianScene(Scene):
class VariationalAutoencoderScene(Scene):
def construct(self):
embedding_layer = EmbeddingLayer()