shap.datasets.nhanesi

shap.datasets.nhanesi(display=False, n_points=None)[源代码]

返回一个包含NHANES I数据的精美打包版本,以生存时间作为标签。

参数:
显示bool, 可选

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

n_pointsint, 可选

要采样的数据点数量。默认是 None(返回整个数据集)。

返回:
如果 display 为 True,则返回特征的修改版本以供显示,同时返回生存时间。
如果 display 为 False,则返回原始特征以及生存时间。

示例

使用示例:

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