prakharg24 commited on
Commit
0539809
·
verified ·
1 Parent(s): f0b737f

Update my_pages/multiverse.py

Browse files
Files changed (1) hide show
  1. 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, max_iter=iterations)}
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)