mirror of
https://github.com/helblazer811/ManimML.git
synced 2025-08-06 17:29:45 +08:00
16 lines
582 B
Python
16 lines
582 B
Python
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
|
|
|
|
class FeedForwardNeuralNetwork(NeuralNetwork):
|
|
"""NeuralNetwork with just feed forward layers"""
|
|
|
|
def __init__(self, layer_node_count, node_radius=0.08, node_color=BLUE, **kwargs):
|
|
# construct layers
|
|
layers = []
|
|
for num_nodes in layer_node_count:
|
|
layer = FeedForwardLayer(
|
|
num_nodes, node_color=node_color, node_radius=node_radius
|
|
)
|
|
layers.append(layer)
|
|
# call super class
|
|
super().__init__(layers, **kwargs)
|