opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-05-17 18:55:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Finished oracle guidance video. Integrated various changes necessary to complete this.

Browse Source
This commit is contained in:
Alec Helbling
2022-04-28 01:37:52 -04:00
parent 95a36eb234
commit 9310b48c56
38 changed files with 1039 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
tests/test_embedding_layer.py
View File

@ -26,6 +26,52 @@ class EmbeddingNNScene(Scene):
self.play(neural_network.make_forward_pass_animation(run_time=5))
class TripletEmbeddingNNScene(Scene):
def construct(self):
embedding_layer = EmbeddingLayer()
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5)
])
self.play(Create(neural_network))
self.play(
neural_network.make_forward_pass_animation(
layer_args={
embedding_layer: {
"triplet_args": {
"anchor_dist": {
"cov": np.array([[0.3, 0], [0, 0.3]]),
"mean": np.array([0.7, 1.4]),
"dist_theme": "ellipse",
"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
},
"negative_dist": {
"cov": np.array([[0.4, 0], [0, 0.25]]),
"mean": np.array([-1, -1.2]),
"dist_theme": "ellipse",
"color": RED
}
}
}
},
run_time=5
)
)
class QueryEmbeddingNNScene(Scene):
def construct(self):
@ -47,7 +93,18 @@ class QueryEmbeddingNNScene(Scene):
run_time=5,
layer_args={
embedding_layer: {
"query_locations": (np.array([2, 2]), np.array([1, 1]))
"positive_dist_args": {
"cov": np.array([[1, 0], [0, 1]]),
"mean": np.array([1, 1]),
"dist_theme": "ellipse",
"color": GREEN
},
"negative_dist_args": {
"cov": np.array([[1, 0], [0, 1]]),
"mean": np.array([-1, -1]),
"dist_theme": "ellipse",
"color": RED
}
}
}
)

19
tests/test_neural_network.py
View File

@ -94,7 +94,7 @@ class NeuralNetworkScene(Scene):
def construct(self):
# Make the Layer object
layers = [
FeedForwardLayer(3, title="Title Test"),
FeedForwardLayer(3),
FeedForwardLayer(5),
FeedForwardLayer(3)
]
@ -151,23 +151,6 @@ class ImageNeuralNetworkScene(Scene):
self.play(nn.make_forward_pass_animation(run_time=5))
self.play(nn.make_forward_pass_animation(run_time=5))
class EmbeddingNNScene(Scene):
def construct(self):
embedding_layer = EmbeddingLayer()
neural_network = NeuralNetwork([
FeedForwardLayer(5),
FeedForwardLayer(3),
embedding_layer,
FeedForwardLayer(3),
FeedForwardLayer(5)
])
self.play(Create(neural_network))
self.play(neural_network.make_forward_pass_animation(run_time=5))
class RecursiveNNScene(Scene):
def construct(self):