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

75
tests/test_embedding_layer.py
View File

@ -4,23 +4,25 @@ from manim_ml.neural_network.layers.embedding import EmbeddingLayer
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
from manim_ml.neural_network.neural_network import NeuralNetwork
config.pixel_height = 720
config.pixel_width = 1280
config.pixel_height = 720
config.pixel_width = 1280
config.frame_height = 5.0
config.frame_width = 5.0
class EmbeddingNNScene(Scene):
class EmbeddingNNScene(Scene):
def construct(self):
embedding_layer = EmbeddingLayer()
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5)
])
neural_network = NeuralNetwork(
[
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5),
]
)
self.play(Create(neural_network))
@ -28,17 +30,18 @@ class EmbeddingNNScene(Scene):
class TripletEmbeddingNNScene(Scene):
def construct(self):
embedding_layer = EmbeddingLayer()
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5)
])
neural_network = NeuralNetwork(
[
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5),
]
)
self.play(Create(neural_network))
@ -51,40 +54,42 @@ class TripletEmbeddingNNScene(Scene):
"cov": np.array([[0.3, 0], [0, 0.3]]),
"mean": np.array([0.7, 1.4]),
"dist_theme": "ellipse",
"color": BLUE
"color": BLUE,
},
"positive_dist": {
"cov": np.array([[0.2, 0], [0, 0.2]]),
"mean": np.array([0.8, -0.4]),
"dist_theme": "ellipse",
"color": GREEN
"color": GREEN,
},
"negative_dist": {
"cov": np.array([[0.4, 0], [0, 0.25]]),
"mean": np.array([-1, -1.2]),
"dist_theme": "ellipse",
"color": RED
}
"color": RED,
},
}
}
},
run_time=5
run_time=5,
)
)
class QueryEmbeddingNNScene(Scene):
class QueryEmbeddingNNScene(Scene):
def construct(self):
embedding_layer = EmbeddingLayer()
embedding_layer.paired_query_mode = True
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5)
])
neural_network = NeuralNetwork(
[
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5),
]
)
self.play(Create(neural_network), run_time=2)
@ -97,15 +102,15 @@ class QueryEmbeddingNNScene(Scene):
"cov": np.array([[1, 0], [0, 1]]),
"mean": np.array([1, 1]),
"dist_theme": "ellipse",
"color": GREEN
"color": GREEN,
},
"negative_dist_args": {
"cov": np.array([[1, 0], [0, 1]]),
"mean": np.array([-1, -1]),
"dist_theme": "ellipse",
"color": RED
}
"color": RED,
},
}
}
},
)
)