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

Used Black to reformat the code in the repository.

Browse Source
This commit is contained in:
Alec Helbling
2023-01-01 23:24:59 -05:00
parent 334662e8c8
commit 3d6e8072e1
71 changed files with 1701 additions and 1135 deletions
Download Patch File Download Diff File Expand all files Collapse all files

117
tests/test_neural_network.py
View File

@ -1,12 +1,21 @@
from cv2 import exp
from manim import *
from manim_ml.neural_network.layers.embedding import EmbeddingLayer
from manim_ml.neural_network.layers.embedding_to_feed_forward import EmbeddingToFeedForward
from manim_ml.neural_network.layers.embedding_to_feed_forward import (
EmbeddingToFeedForward,
)
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
from manim_ml.neural_network.layers.feed_forward_to_embedding import FeedForwardToEmbedding
from manim_ml.neural_network.layers.feed_forward_to_feed_forward import FeedForwardToFeedForward
from manim_ml.neural_network.layers.feed_forward_to_embedding import (
FeedForwardToEmbedding,
)
from manim_ml.neural_network.layers.feed_forward_to_feed_forward import (
FeedForwardToFeedForward,
)
from manim_ml.neural_network.layers.image import ImageLayer
from manim_ml.neural_network.neural_network import NeuralNetwork, FeedForwardNeuralNetwork
from manim_ml.neural_network.neural_network import (
NeuralNetwork,
FeedForwardNeuralNetwork,
)
from PIL import Image
import numpy as np
@ -19,28 +28,30 @@ config.frame_width = 6.0
Unit Tests
"""
def assert_classes_match(all_layers, expected_classes):
assert len(list(all_layers)) == 5
for index, layer in enumerate(all_layers):
expected_class = expected_classes[index]
assert isinstance(layer, expected_class), f"Wrong layer class {layer.__class__} expected {expected_class}"
assert isinstance(
layer, expected_class
), f"Wrong layer class {layer.__class__} expected {expected_class}"
def test_embedding_layer():
embedding_layer = EmbeddingLayer()
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer
])
neural_network = NeuralNetwork(
[FeedForwardLayer(5), FeedForwardLayer(3), embedding_layer]
)
expected_classes = [
FeedForwardLayer,
FeedForwardLayer,
FeedForwardToFeedForward,
FeedForwardLayer,
FeedForwardToEmbedding,
EmbeddingLayer
EmbeddingLayer,
]
assert_classes_match(neural_network.all_layers, expected_classes)
@ -49,18 +60,16 @@ def test_embedding_layer():
def test_remove_layer():
embedding_layer = EmbeddingLayer()
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer
])
neural_network = NeuralNetwork(
[FeedForwardLayer(5), FeedForwardLayer(3), embedding_layer]
)
expected_classes = [
FeedForwardLayer,
FeedForwardLayer,
FeedForwardToFeedForward,
FeedForwardLayer,
FeedForwardToEmbedding,
EmbeddingLayer
EmbeddingLayer,
]
assert_classes_match(neural_network.all_layers, expected_classes)
@ -72,7 +81,7 @@ def test_remove_layer():
print(list(neural_network.all_layers))
expected_classes = [
FeedForwardLayer,
FeedForwardLayer,
FeedForwardToFeedForward,
FeedForwardLayer,
]
@ -81,36 +90,35 @@ def test_remove_layer():
assert_classes_match(neural_network.all_layers, expected_classes)
class FeedForwardNeuralNetworkScene(Scene):
class FeedForwardNeuralNetworkScene(Scene):
def construct(self):
nn = FeedForwardNeuralNetwork([3, 5, 3])
self.play(Create(nn))
self.play(Wait(3))
class NeuralNetworkScene(Scene):
"""Test Scene for the Neural Network"""
def construct(self):
# Make the Layer object
layers = [
FeedForwardLayer(3),
FeedForwardLayer(5),
FeedForwardLayer(3)
]
layers = [FeedForwardLayer(3), FeedForwardLayer(5), FeedForwardLayer(3)]
nn = NeuralNetwork(layers)
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(Create(nn))
forward_propagation_animation = nn.make_forward_pass_animation(
run_time=5, passing_flash=True
)
self.play(forward_propagation_animation)
class GrayscaleImageNeuralNetworkScene(Scene):
class GrayscaleImageNeuralNetworkScene(Scene):
def construct(self):
image = Image.open('images/image.jpeg')
image = Image.open("images/image.jpeg")
numpy_image = np.asarray(image)
# Make nn
layers = [
@ -118,7 +126,7 @@ class GrayscaleImageNeuralNetworkScene(Scene):
FeedForwardLayer(5),
FeedForwardLayer(3),
FeedForwardLayer(6),
ImageLayer(numpy_image, height=1.4)
ImageLayer(numpy_image, height=1.4),
]
nn = NeuralNetwork(layers)
nn.scale(1.3)
@ -129,10 +137,10 @@ class GrayscaleImageNeuralNetworkScene(Scene):
self.play(nn.make_forward_pass_animation(run_time=5))
self.play(nn.make_forward_pass_animation(run_time=5))
class ImageNeuralNetworkScene(Scene):
def construct(self):
image = Image.open('../assets/gan/real_image.jpg')
image = Image.open("../assets/gan/real_image.jpg")
numpy_image = np.asarray(image)
# Make nn
layers = [
@ -140,7 +148,7 @@ class ImageNeuralNetworkScene(Scene):
FeedForwardLayer(5),
FeedForwardLayer(3),
FeedForwardLayer(6),
ImageLayer(numpy_image, height=1.4)
ImageLayer(numpy_image, height=1.4),
]
nn = NeuralNetwork(layers)
nn.scale(1.3)
@ -151,44 +159,40 @@ class ImageNeuralNetworkScene(Scene):
self.play(nn.make_forward_pass_animation(run_time=5))
self.play(nn.make_forward_pass_animation(run_time=5))
class RecursiveNNScene(Scene):
class RecursiveNNScene(Scene):
def construct(self):
nn = NeuralNetwork([
NeuralNetwork([
FeedForwardLayer(3),
FeedForwardLayer(2)
]),
NeuralNetwork([
FeedForwardLayer(2),
FeedForwardLayer(3)
])
])
nn = NeuralNetwork(
[
NeuralNetwork([FeedForwardLayer(3), FeedForwardLayer(2)]),
NeuralNetwork([FeedForwardLayer(2), FeedForwardLayer(3)]),
]
)
self.play(Create(nn))
class LayerInsertionScene(Scene):
class LayerInsertionScene(Scene):
def construct(self):
pass
class LayerRemovalScene(Scene):
class LayerRemovalScene(Scene):
def construct(self):
image = Image.open('images/image.jpeg')
image = Image.open("images/image.jpeg")
numpy_image = np.asarray(image)
layer = FeedForwardLayer(5)
layers = [
ImageLayer(numpy_image, height=1.4),
FeedForwardLayer(3),
FeedForwardLayer(3),
layer,
FeedForwardLayer(3),
FeedForwardLayer(6)
FeedForwardLayer(6),
]
nn = NeuralNetwork(layers)
self.play(Create(nn))
remove_animation = nn.remove_layer(layer)
print("before remove")
@ -196,21 +200,21 @@ class LayerRemovalScene(Scene):
print(nn)
print("after remove")
class LayerInsertionScene(Scene):
class LayerInsertionScene(Scene):
def construct(self):
image = Image.open('images/image.jpeg')
image = Image.open("images/image.jpeg")
numpy_image = np.asarray(image)
layers = [
ImageLayer(numpy_image, height=1.4),
FeedForwardLayer(3),
FeedForwardLayer(3),
FeedForwardLayer(6)
FeedForwardLayer(6),
]
nn = NeuralNetwork(layers)
self.play(Create(nn))
layer = FeedForwardLayer(5)
@ -219,11 +223,12 @@ class LayerInsertionScene(Scene):
print(nn)
print("after remove")
if __name__ == "__main__":
"""Render all scenes"""
# Feed Forward Neural Network
ffnn_scene = FeedForwardNeuralNetworkScene()
ffnn_scene.render()
# Neural Network
# Neural Network
nn_scene = NeuralNetworkScene()
nn_scene.render()