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

Working initial visualization of a static decision tree.

Browse Source
This commit is contained in:
Alec Helbling
2023-01-04 20:50:01 -05:00
parent 1d9cad8587
commit e33f98373a
6 changed files with 295 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
manim_ml/neural_network/neural_network.py
View File

@ -170,7 +170,8 @@ class NeuralNetwork(Group):
after_layer_args = {}
if layer.input_layer in layer_args:
before_layer_args = layer_args[layer.input_layer]
current_layer_args = layer_args[layer]
if layer in layer_args:
current_layer_args = layer_args[layer]
if layer.output_layer in layer_args:
after_layer_args = layer_args[layer.output_layer]
# Merge the two dicts