konstantinG commited on
Commit
233fa77
1 Parent(s): 92461c6

Update classification.py

Browse files
Files changed (1) hide show
  1. classification.py +1 -1
classification.py CHANGED
@@ -10,7 +10,7 @@ def prep_and_train(targ, data, models=None):
10
  return best, s_df, best_df
11
 
12
 
13
- @st.cache_resource
14
  def tuning(_model, n_iters, search_lib):
15
  tuned_dt = tune_model(estimator=_model, n_iter=n_iters, search_library=search_lib, choose_better=True)
16
  info_df = pull()
 
10
  return best, s_df, best_df
11
 
12
 
13
+ @st.experimental_memo
14
  def tuning(_model, n_iters, search_lib):
15
  tuned_dt = tune_model(estimator=_model, n_iter=n_iters, search_library=search_lib, choose_better=True)
16
  info_df = pull()