From 99d943daac98438bd12a973edd87944f38961785 Mon Sep 17 00:00:00 2001 From: CharlesCousyn Date: Fri, 31 Jan 2025 08:18:41 -0500 Subject: [PATCH] Correction using n_interpolation_size instead of _n_features_in_tree --- shapiq/explainer/tree/treeshapiq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/shapiq/explainer/tree/treeshapiq.py b/shapiq/explainer/tree/treeshapiq.py index 90b1eccf..221f495f 100644 --- a/shapiq/explainer/tree/treeshapiq.py +++ b/shapiq/explainer/tree/treeshapiq.py @@ -409,7 +409,7 @@ def _init_summary_polynomials(self): self.subset_ancestors_store[order] = subset_ancestors # If the tree has only one feature, we assign a default value of 0 - if self._n_features_in_tree == 1: + if self.n_interpolation_size == 1: self.D_store[order] = np.array([0]) else: self.D_store[order] = np.polynomial.chebyshev.chebpts2(self.n_interpolation_size)