Added ListGroup class for better management of a group of objects.

This commit is contained in:
Alec Helbling
2022-04-18 23:56:31 -04:00
parent 0febbe547d
commit 05f512f185
19 changed files with 424 additions and 71 deletions

View File

@ -5,8 +5,8 @@ from manim_ml.neural_network.layers.parent_layers import VGroupNeuralNetworkLaye
class EmbeddingLayer(VGroupNeuralNetworkLayer):
"""NeuralNetwork embedding object that can show probability distributions"""
def __init__(self, point_radius=0.02):
super(EmbeddingLayer, self).__init__()
def __init__(self, point_radius=0.02, **kwargs):
super(EmbeddingLayer, self).__init__(**kwargs)
self.point_radius = point_radius
self.axes = Axes(
tips=False,