Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -264,8 +264,8 @@ def render():
         X_test = pca.transform(X_test)
     
     placeholder = st.empty()
-    modelclass_dict = {'Neural Network (Small)': MLPClassifier([10], random_state=seed, max_iter=iterations),
-                       'Logistic Regression': SGDClassifier(random_state=seed, max_iter=iterations),
+    modelclass_dict = {'Neural Network (Small)': MLPClassifier([10], random_state=seed, max_iter=500),
+                       'Logistic Regression': SGDClassifier(random_state=seed, max_iter=500),
                        'Decision Tree': DecisionTreeClassifier(random_state=seed)}
     model = modelclass_dict[arch]
     placeholder.write("Training your model.")