shap.datasets.linnerud

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

返回 Linnerud 数据集,以方便用于多目标回归。

参数:
n_pointsint, 可选

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

返回值:
Xpd.DataFrame

特征数据。

ypd.DataFrame

多类别目标变量。

注释

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

  • 特征矩阵 X 包括三个运动变量:ChinsSitupsJumps

  • 目标变量 y 包括三个生理测量值:WeightWaistPulse

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

示例

获取特征矩阵和目标变量

features, targets = shap.datasets.linnerud()

获取数据子集

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