CosmickVisions commited on
Commit
3ca12d9
·
verified ·
1 Parent(s): bebe6ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,10 +53,10 @@ elif app_mode == "Model Training":
53
  if st.button("Setup PyCaret"):
54
  with st.spinner("Setting up PyCaret..."):
55
  if problem_type == "Classification":
56
- setup(data=df, target=target, session_id=123, verbose=False)
57
  st.session_state.problem_type = "Classification"
58
  elif problem_type == "Regression":
59
- setup(data=df, target=target, session_id=123, verbose=False)
60
  st.session_state.problem_type = "Regression"
61
  elif problem_type == "Clustering":
62
  setup(data=df, session_id=123, verbose=False)
 
53
  if st.button("Setup PyCaret"):
54
  with st.spinner("Setting up PyCaret..."):
55
  if problem_type == "Classification":
56
+ setup(data=df, session_id=123, verbose=False)
57
  st.session_state.problem_type = "Classification"
58
  elif problem_type == "Regression":
59
+ setup(data=df, session_id=123, verbose=False)
60
  st.session_state.problem_type = "Regression"
61
  elif problem_type == "Clustering":
62
  setup(data=df, session_id=123, verbose=False)