shap.datasets.linnerud

shap.datasets.linnerud(n_points: int | None = None) → tuple[DataFrame, DataFrame]

以方便多目标回归的包形式返回 Linnerud 数据集。

参数:

n_pointsint, 可选

要采样的数据点数量。如果提供，则随机采样指定数量的数据点。

返回:

Xpd.DataFrame

特征数据。

ypd.DataFrame

多类别目标变量。

备注

• Linnerud 数据集包含 20 个人的生理和运动数据。

• 特征矩阵 X 包含三个运动变量：Chins（引体向上）、Situps（仰卧起坐）、Jumps（跳跃）。

• 目标变量 y 包含三个生理测量值：Weight（体重）、Waist（腰围）、Pulse（脉搏）。

更多详情：sklearn.datasets.load_linnerud()

示例

获取特征矩阵和目标变量

features, targets = shap.datasets.linnerud()

获取数据子集

subset_features, subset_targets = shap.datasets.linnerud(n_points=100)