shap.utils.partition_tree

shap.utils.partition_tree(X, metric='correlation')