PartitionExplainer with non-binary partition_tree

[IMW112]

Problem Description

Hi,

I would like to use shap to compute Owen values where partition_tree describes an arbitrary coalition structure, e.g., imagine we had three different 'teams' at the first level of the hierarchy, and each team had a variable number of 'players' at the second level.

I have been looking at providing a custom partition_tree array to PartitionExplainer to achieve this, but, as far as I have been able to tell, there is no way to describe a coalition structure other than a binary tree (e.g., a dendrogram resulting from cluster analysis).

Is this a feature that could be added please, or is there some other workaround?

Many thanks

Alternative Solutions

No response

Additional Context

No response

Feature request checklist

• I have checked the issue tracker for duplicate issues.
• I'd be interested in making a PR to implement this feature