konstantinG commited on
Commit
92461c6
·
1 Parent(s): 0544226

Update classification.py

Browse files
Files changed (1) hide show
  1. classification.py +1 -1
classification.py CHANGED
@@ -1,7 +1,7 @@
1
  from pycaret.classification import *
2
  import streamlit as st
3
 
4
- @st.cache_data
5
  def prep_and_train(targ, data, models=None):
6
  s = setup(data, target = targ, session_id = 12)
7
  s_df = pull()
 
1
  from pycaret.classification import *
2
  import streamlit as st
3
 
4
+ @st.experimental_memo
5
  def prep_and_train(targ, data, models=None):
6
  s = setup(data, target = targ, session_id = 12)
7
  s_df = pull()