opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-08-23 10:01:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Changed default functionality of activation functions so they are all lined up with eachother.

Browse Source
This commit is contained in:
Alec Helbling
2023-04-09 20:15:07 -04:00
parent 2a50124ae2
commit ca7978929a
13 changed files with 397 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
manim_ml/neural_network/activation_functions/activation_function.py
View File

@ -3,7 +3,7 @@ from abc import ABC, abstractmethod
import random
import manim_ml.neural_network.activation_functions.relu as relu
import manim_ml
class ActivationFunction(ABC, VGroup):
"""Abstract parent class for defining activation functions"""
@ -16,9 +16,9 @@ class ActivationFunction(ABC, VGroup):
x_length=0.5,
y_length=0.3,
show_function_name=True,
active_color=ORANGE,
plot_color=BLUE,
rectangle_color=WHITE,
active_color=manim_ml.config.color_scheme.active_color,
plot_color=manim_ml.config.color_scheme.primary_color,
rectangle_color=manim_ml.config.color_scheme.secondary_color,
):
super(VGroup, self).__init__()
self.function_name = function_name
@ -46,6 +46,7 @@ class ActivationFunction(ABC, VGroup):
"include_numbers": False,
"stroke_width": 0.5,
"include_ticks": False,
"color": self.rectangle_color
},
)
self.add(self.axes)