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