Change classifier

XAITK · Sep 22, 2021 · 8c3b31e · 8c3b31e
1 parent ebcc7b6
commit 8c3b31e
Showing 1 changed file with 16 additions and 16 deletions.
diff --git a/cml_example/train.py b/cml_example/train.py
@@ -46,23 +46,23 @@
 ###############################################################################
 # Train a scikit-learn estimator model -- assuming as .predict_proba
 print("Training model")
-model = LogisticRegression(
-    C=50. / train_samples,
-    penalty='l1',
-    solver='saga',
-    tol=0.1,
-    # important to make model training some sort of deterministic
-    random_state=0,
-)
-# model = MLPClassifier(
-#     hidden_layer_sizes=(50,),
-#     # max_iter=10,
-#     alpha=1e-4,
-#     solver='sgd',
-#     verbose=10,
-#     random_state=1,
-#     learning_rate_init=.1
+# model = LogisticRegression(
+#     C=50. / train_samples,
+#     penalty='l1',
+#     solver='saga',
+#     tol=0.1,
+#     # important to make model training some sort of deterministic
+#     random_state=0,
 # )
+model = MLPClassifier(
+    hidden_layer_sizes=(50,),
+    # max_iter=10,
+    alpha=1e-4,
+    solver='sgd',
+    verbose=10,
+    random_state=1,
+    learning_rate_init=.1
+)
 model.fit(X_train, y_train)
 print("Training model -- Done")