Bugfix: Plausible counterfactuals of a decision tree classifier - fix…

… number of features
andreArtelt · Jun 3, 2020 · 12f2782 · 12f2782
1 parent ee4e87c
commit 12f2782
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ceml/sklearn/decisiontree.py b/ceml/sklearn/decisiontree.py
@@ -150,7 +150,7 @@ class DecisionTreeCounterfactual(SklearnCounterfactual, PlausibleCounterfactualO
     See parent class :class:`ceml.sklearn.counterfactual.SklearnCounterfactual`.
     """
     def __init__(self, model, **kwds):
-        super().__init__(model=model, tree_model=model, n_dims=None, **kwds)
+        super().__init__(model=model, tree_model=model, n_dims=model.n_features_, **kwds)
 
     def rebuild_model(self, model):
         """Rebuild a :class:`sklearn.linear_model.LogisticRegression` model.