shap.utils.hclust

shap.utils.hclust(X: _ArrayLike, y: _ArrayLike | None = None, linkage: Literal['single', 'complete', 'average'] = 'single', metric: str = 'auto', random_state: int | np.random.RandomState = 0) → np.ndarray

针对目标变量 y，为特征 X 拟合一个层次聚类模型。

有关聚类方法的更多信息，请参阅 scipy.cluster.hierarchy.linkage()。

有关 scipy 距离度量的更多信息，请参阅 scipy.spatial.distance.pdist()。

参数:

X: 2d-array-like（二维类数组）

要聚类的特征

y: array-like（类数组）或 None

目标变量

linkage: str（字符串）

定义计算簇之间距离的方法。必须是“single”、“complete”或“average”之一。

metric: str（字符串）

Scipy 距离度量或 “xgboost_distances_r2”。

• 如果为 xgboost_distances_r2，则使用 shap.utils.xgboost_distances_r2() 来估计特征 X 之间相对于目标变量 y 的冗余距离。

• 否则，使用给定的距离度量计算特征之间的距离。

• 如果为 auto（默认值），则在提供了目标变量时使用 xgboost_distances_r2，否则使用 cosine 距离度量。

random_state: int（整数）或 np.random.RandomState

Numpy 随机状态，默认为 0。

返回:

clustering: np.array

编码为链接矩阵的层次聚类。