Update my_pages/multiverse.py
Browse files- my_pages/multiverse.py +1 -1
my_pages/multiverse.py
CHANGED
|
@@ -258,7 +258,7 @@ def render():
|
|
| 258 |
placeholder = st.empty()
|
| 259 |
modelclass_dict = {'Neural Network (Small)': MLPClassifier([10], random_state=random_seed, max_iter=iterations),
|
| 260 |
'Logistic Regression': SGDClassifier(random_state=random_seed, max_iter=iterations),
|
| 261 |
-
'Random Forest': RandomForestClassifier(random_state=random_seed
|
| 262 |
model = modelclass_dict[arch]
|
| 263 |
placeholder.write("Training your model.")
|
| 264 |
model.fit(X_train, y_train)
|
|
|
|
| 258 |
placeholder = st.empty()
|
| 259 |
modelclass_dict = {'Neural Network (Small)': MLPClassifier([10], random_state=random_seed, max_iter=iterations),
|
| 260 |
'Logistic Regression': SGDClassifier(random_state=random_seed, max_iter=iterations),
|
| 261 |
+
'Random Forest': RandomForestClassifier(random_state=random_seed)}
|
| 262 |
model = modelclass_dict[arch]
|
| 263 |
placeholder.write("Training your model.")
|
| 264 |
model.fit(X_train, y_train)
|