diff --git a/scripts/precompute/shap_analysis.py b/scripts/precompute/shap_analysis.py index 87d9f020..f1ede2f1 100644 --- a/scripts/precompute/shap_analysis.py +++ b/scripts/precompute/shap_analysis.py @@ -85,7 +85,8 @@ def main(): model.predict, shap.maskers.Independent(background, max_samples=50), ) - shap_values = explainer(X_shap[:min(200, len(X_shap))]).values + max_evals = 2 * X_shap.shape[1] + 1 + shap_values = explainer(X_shap[:min(200, len(X_shap))], max_evals=max_evals).values # Get feature names feature_names = preprocessor.get_feature_names_out()