shap.utils.partition_tree_shuffle

shap.utils.partition_tree_shuffle(indexes: ndarray[Any, dtype[Any]], index_mask: ndarray[Any, dtype[bool]], partition_tree: ndarray[Any, dtype[Any]]) → None

以与给定分区树一致的方式随机打乱索引。

参数:

indexes: np.array

我们要打乱的索引的输出位置。请注意，len(indexes) 应等于 index_mask.sum()。

index_mask: np.array

一个布尔掩码，指示我们要包含在打乱列表中的索引。

partition_tree: np.array

我们应该遵循的分区树。