Skip to content

Commit

Permalink
Modifty comments
Browse files Browse the repository at this point in the history
  • Loading branch information
CharlesCousyn committed Jan 30, 2025
1 parent 3e96dae commit e8c961d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions shapiq/explainer/tree/treeshapiq.py
Original file line number Diff line number Diff line change
Expand Up @@ -408,9 +408,9 @@ def _init_summary_polynomials(self):
)
self.subset_ancestors_store[order] = subset_ancestors

# If the tree has only one feature
# If the tree has only one feature, we assign a default value of 0
if self._n_features_in_tree == 1:
self.D_store[order] = np.array([0]) # Assign a default value
self.D_store[order] = np.array([0])
else:
self.D_store[order] = np.polynomial.chebyshev.chebpts2(self.n_interpolation_size)

Expand Down

0 comments on commit e8c961d

Please sign in to comment.