File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -654,15 +654,14 @@ def predict(example):
654
654
# ______________________________________________________________________________
655
655
656
656
657
- def NeuralNetLearner (dataset , hidden_layer_sizes = None ,
657
+ def NeuralNetLearner (dataset , hidden_layer_sizes = [ 3 ] ,
658
658
learning_rate = 0.01 , epochs = 100 , activation = sigmoid ):
659
659
"""Layered feed-forward network.
660
660
hidden_layer_sizes: List of number of hidden units per hidden layer
661
661
learning_rate: Learning rate of gradient descent
662
662
epochs: Number of passes over the dataset
663
663
"""
664
664
665
- hidden_layer_sizes = hidden_layer_sizes or [3 ] # default value
666
665
i_units = len (dataset .inputs )
667
666
o_units = len (dataset .values [dataset .target ])
668
667
You can’t perform that action at this time.
0 commit comments