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)