From ea1ab9fe4ffe95de7af9fa0f9e69120fda37c3c4 Mon Sep 17 00:00:00 2001
From: Elle O'Brien <elleobrien11@gmail.com>
Date: Tue, 16 Feb 2021 11:42:04 -0800
Subject: [PATCH] Update train.py

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index b8cc6e6..d8af4d9 100644
--- a/train.py
+++ b/train.py
@@ -15,7 +15,7 @@
 
 # Fit a model
 
-clf = MLPClassifier(random_state=0, max_iter=30)
+clf = MLPClassifier(random_state=0, max_iter=30, solver="sgd")
 clf.fit(X_train,y_train)
 
 # Get overall accuracy