small changes in indices selection

scikit-learn-contrib · Apr 15, 2024 · 4ab46ad · 4ab46ad
1 parent 73f74f6
commit 4ab46ad
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/docs/examples/plot_lcppn_explainer.py b/docs/examples/plot_lcppn_explainer.py
@@ -33,12 +33,11 @@
 explanations = explainer.explain(X_test.values)
 print(explanations)
 
-# Filter samples which only predicted "Respiratory"
-respiratory_idx = explainer.get_sample_indices(predictions, "Respiratory")
-
 # Use .sel() method to apply the filter and obtain filtered results
 shap_val_respiratory = explainer.filter_by_class(
-    explanations, class_name="Respiratory", sample_indices=respiratory_idx
+    explanations,
+    class_name="Respiratory",
+    sample_indices=explainer.get_sample_indices(predictions, "Respiratory"),
 )