Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
| 178 |
np.random.seed(seed)
|
| 179 |
-
preproc =
|
| 180 |
-
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:
|