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

Fixed bug with scaling title.

Browse Source
This commit is contained in:
Alec Helbling
2023-04-13 19:30:21 -04:00
parent ca7978929a
commit 775d597517
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
manim_ml/neural_network/neural_network.py
View File

@ -62,7 +62,10 @@ class NeuralNetwork(Group):
# Place the connective layers
self._place_connective_layers()
# Make overhead title
self.title = Text(self.title_text, font_size=DEFAULT_FONT_SIZE / 2)
self.title = Text(
self.title_text,
font_size=DEFAULT_FONT_SIZE / 2
)
self.title.next_to(self, UP, 1.0)
self.add(self.title)
# Place layers at correct z index
@ -190,6 +193,9 @@ class NeuralNetwork(Group):
current_layer.activation_function.move_to(
current_layer,
)
current_layer.activation_function.match_y(
self.input_layers[0]
)
current_layer.activation_function.shift(
up_movement
)
@ -364,6 +370,8 @@ class NeuralNetwork(Group):
def scale(self, scale_factor, **kwargs):
"""Overriden scale"""
prior_center = self.get_center()
for layer in self.all_layers:
layer.scale(scale_factor, **kwargs)
# Place layers with scaled spacing
@ -371,6 +379,13 @@ class NeuralNetwork(Group):
# self.connective_layers, self.all_layers = self._construct_connective_layers()
self._place_layers(layout=self.layout, layout_direction=self.layout_direction)
self._place_connective_layers()
# Scale the title
self.remove(self.title)
self.title.scale(scale_factor, **kwargs)
self.title.next_to(self, UP * scale_factor, buff=1.0 * scale_factor)
self.add(self.title)
self.move_to(prior_center)
def filter_layers(self, function):
"""Filters layers of the network given function"""