changing mlp hyperp

Kotzly · Aug 20, 2022 · 8624865 · 8624865
1 parent 4fd07d3
commit 8624865
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/experiments/issue_12/utils.py b/experiments/issue_12/utils.py
@@ -291,16 +291,13 @@ def get_classifier(clf_method, n_inputs=10, random_state=1):
         clf = GaussianNB()
         param_grid = dict()
     elif clf_method == "mlp":
-        clf = MLPClassifier(random_state=random_state, max_iter=5000)
+        clf = MLPClassifier(validation_fraction=0.2, random_state=random_state, max_iter=2500)
         param_grid = dict(
             hidden_layer_sizes=[
                 (),
-                (n_inputs,),
                 (n_inputs // 2,),
                 (n_inputs // 4),
-                (n_inputs // 2, n_inputs // 2),
                 (n_inputs // 4, n_inputs // 4),
-                (n_inputs // 2, n_inputs // 2, n_inputs // 2),
             ],
             activation=["relu", "logistic"]
         )