shap.datasets.nhanesi

shap.datasets.nhanesi(display=False, n_points=None)

Return a nicely packaged version of NHANES I data with survival times as labels.

Parameters:

displaybool, optional

If True, returns the features with a modified display. Default is False.

n_pointsint, optional

Number of data points to sample. Default is None (returns the entire dataset).

Returns:

If display is True, returns a modified version of the features for display along with survival times.

If display is False, returns the original features along with survival times.

Examples

Usage example:

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