Skip to content

Feature/mobius for trees - #596

Open
ron-wettenstein wants to merge 8 commits into
mmschlk:mainfrom
ron-wettenstein:feature/mobius_for_trees
Open

Feature/mobius for trees#596
ron-wettenstein wants to merge 8 commits into
mmschlk:mainfrom
ron-wettenstein:feature/mobius_for_trees

Conversation

@ron-wettenstein

@ron-wettenstein ron-wettenstein commented Sep 1, 2026

Copy link
Copy Markdown
Contributor

Motivation and Context

TreeExplainer now supports also the "Mobius" index.
Notes:

  1. This branch was created from the feature/woodelf_stii_fsii_fbii branch. Both touch similar parts of the code and I wanted to avoid conflicts. So this PR include also the code added in Feature/woodelf stii fsii fbii #595. We should first merge the Feature/woodelf stii fsii fbii #595 PR and then review this PR.
  2. I added Mobius to the supported indices of TreeExplainer. It is not part of the supported indices in the base Explainer class though. I do some casting when calling the super init. It this the right behaviour? Do we want to add Mobius to the base Exaplainer class, and maybe use the opportunity to support in in more explainers (It seems that in TabularExplainer Mobius will be easy to compute exactly but hard to estimate, lets talk about it).

Public API Changes

  • No Public API changes
  • Yes, Public API changes (Details below)

TreeExplainer now support index="Mobius"


How Has This Been Tested?

Manually and with tests. The test check the QuadratureTreeSHAP and IntervantaionalTreeSHAP implementation and compare it to Woodelf.


Checklist

  • The changes have been tested locally.
  • Documentation has been updated (if the public API or usage changes).
  • An entry has been added to CHANGELOG.md (if relevant for users).
  • The code follows the project's style guidelines.
  • I have considered the impact of these changes on the public API.

@codecov

codecov Bot commented Sep 1, 2026

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant