Skip to content

Add Random Forrest Support for TreeSHAP-IQ #55

Closed
@mmschlk

Description

@mmschlk

Similar to the already existing sklearn decision tree support, we need to also support sklearn random forests naturally.

For this a special mechanism in the conversion of tree-models to their edge representation is required.

Idea: RF average the predictions of the ensemble members. The same goes for their Shapley interactions. A list of EdgeTrees is probably right.

The implementation in the o.g. TreeSHAP-IQ should work.

Tasks

  • Add RF converter
  • Add tests

Metadata

Metadata

Assignees

Labels

feature 💡New feature or enhancement requesthelp wanted 🙏Extra attention is needed

Projects

Status

✅ Done

Relationships

None yet

Development

No branches or pull requests

Issue actions