shapiq.datasets.load_adult_census¶

shapiq.datasets.load_adult_census(*, to_numpy=False)[source]¶

Load the adult census dataset from the UCI Machine Learning Repository.

Original source: https://archive.ics.uci.edu/ml/datasets/adult

Note

The function requires the sklearn package to be installed.

Parameters:: to_numpy (bool) – Return numpy objects instead of pandas. Default is False.
Return type:: tuple[DataFrame, Series | DataFrame] | tuple[ndarray, ndarray]
Returns:: The adult census dataset as a pandas DataFrame.

Example

>>> from shapiq.datasets import load_adult_census
>>> x_data, y_data = load_adult_census()
>>> print(x_data.shape, y_data.shape)
((45222, 14), (45222,))