Skip to content

Commit a4c4c46

Browse files
thesagarsehgalantmarakis
authored andcommitted
Update in NeuralNetLearner function in learnign.py (#1019)
* Update in NeuralNetLearner function * made the changes as suggested
1 parent 9389c38 commit a4c4c46

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

learning.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -654,15 +654,14 @@ def predict(example):
654654
# ______________________________________________________________________________
655655

656656

657-
def NeuralNetLearner(dataset, hidden_layer_sizes=None,
657+
def NeuralNetLearner(dataset, hidden_layer_sizes=[3],
658658
learning_rate=0.01, epochs=100, activation = sigmoid):
659659
"""Layered feed-forward network.
660660
hidden_layer_sizes: List of number of hidden units per hidden layer
661661
learning_rate: Learning rate of gradient descent
662662
epochs: Number of passes over the dataset
663663
"""
664664

665-
hidden_layer_sizes = hidden_layer_sizes or [3] # default value
666665
i_units = len(dataset.inputs)
667666
o_units = len(dataset.values[dataset.target])
668667

0 commit comments

Comments
 (0)