PD03 commited on
Commit
7838063
·
verified ·
1 Parent(s): baaf3da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -174,10 +174,10 @@ def train_model(df: pd.DataFrame, feats_num, feats_cat, target):
174
  return pipe, {"r2": r2_score(y_test, pred), "mae": mean_absolute_error(y_test, pred)}, X_test
175
 
176
  @st.cache_resource(show_spinner=False)
177
- def compute_shap(pipe, X_sample, feats_num, feats_cat, shap_sample=1000, seed=42):
178
  np.random.seed(seed)
179
- preproc = pipe.named_steps["pre"]
180
- rf = pipe.named_steps["rf"]
181
  feature_names = list(preproc.named_transformers_["cat"].get_feature_names_out(feats_cat)) + feats_num
182
 
183
  if len(X_sample) > shap_sample:
 
174
  return pipe, {"r2": r2_score(y_test, pred), "mae": mean_absolute_error(y_test, pred)}, X_test
175
 
176
  @st.cache_resource(show_spinner=False)
177
+ def compute_shap(_pipe, X_sample, feats_num, feats_cat, shap_sample=1000, seed=42):
178
  np.random.seed(seed)
179
+ preproc = _pipe.named_steps["pre"]
180
+ rf = _pipe.named_steps["rf"]
181
  feature_names = list(preproc.named_transformers_["cat"].get_feature_names_out(feats_cat)) + feats_num
182
 
183
  if len(X_sample) > shap_sample: