shap.datasets.nhanesi

shap.datasets.nhanesi(display: bool = False, n_points: int | None = None) → tuple[DataFrame, ndarray]

返回 NHANES I 数据的良好封装版本，并将生存时间作为标签。

用于生存分析任务。

参数:

displaybool, 可选

如果为 True，则返回具有修改后显示的特征。默认为 False。

n_pointsint, 可选

要采样的数据点数量。默认为 None（返回整个数据集）。

返回:

Xpd.DataFrame

特征数据矩阵。如果 display 为 True，则返回修改后的特征版本以供显示，并将其作为 X 返回。

ynp.ndarray

表示生存时间的目标变量。

示例

使用示例

features, survival_times = shap.datasets.nhanesi(display=True, n_points=100)