shap.datasets.linnerud

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

Return the Linnerud dataset in a convenient package for multi-target regression.

Parameters:
n_pointsint, optional

Number of data points to sample. If provided, randomly samples the specified number of points.

Returns:
Xpd.DataFrame

The feature data.

ypd.DataFrame

The multiclass target variables.

Notes

  • The Linnerud dataset contains physiological and exercise data for 20 individuals.

  • The feature matrix X includes three exercise variables: Chins, Situps, Jumps.

  • The target variables y include three physiological measurements: Weight, Waist, Pulse.

More details: sklearn.datasets.load_linnerud()

Examples

To get the feature matrix and target variables:

features, targets = shap.datasets.linnerud()

To get a subset of the data:

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