diff --git a/learning.py b/learning.py index e0d4cd26d..66f13a331 100644 --- a/learning.py +++ b/learning.py @@ -654,7 +654,7 @@ def predict(example): # ______________________________________________________________________________ -def NeuralNetLearner(dataset, hidden_layer_sizes=None, +def NeuralNetLearner(dataset, hidden_layer_sizes=[3], learning_rate=0.01, epochs=100, activation = sigmoid): """Layered feed-forward network. hidden_layer_sizes: List of number of hidden units per hidden layer @@ -662,7 +662,6 @@ def NeuralNetLearner(dataset, hidden_layer_sizes=None, epochs: Number of passes over the dataset """ - hidden_layer_sizes = hidden_layer_sizes or [3] # default value i_units = len(dataset.inputs) o_units = len(dataset.values[dataset.target])