fairlearn.datasets.fetch_diabetes_hospital#

fairlearn.datasets.fetch_diabetes_hospital(*, as_frame=True, cache=True, data_home=None, return_X_y=False)[source]#

Load the preprocessed Diabetes 130-Hospitals dataset (binary classification).

Download it if necessary.

Samples total

101766

Dimensionality

24

Features

numeric, categorical, string

Classes

2

Source: UCI Repository [1] Paper: Strack et al., 2014 [2]

The “Diabetes 130-Hospitals” dataset represents 10 years of clinical care at 130 U.S. hospitals and delivery networks, collected from 1999 to 2008. Each record represents the hospital admission record for a patient diagnosed with diabetes whose stay lasted between one to fourteen days.

The original “Diabetes 130-Hospitals” dataset was collected by Beata Strack, Jonathan P. DeShazo, Chris Gennings, Juan L. Olmo, Sebastian Ventura, Krzysztof J. Cios, and John N. Clore in 2014.

This version of the dataset was derived by the Fairlearn team for the SciPy 2021 tutorial “Fairness in AI Systems: From social context to practice using Fairlearn”. In this version, the target variable “readmitted” is binarized into whether the patient was re-admitted within thirty days. The full pre-processing script is available here.

Read more in the User Guide.

New in version 0.8.0.

Parameters:
as_framebool, default=True

If True, the data is a pandas DataFrame including columns with appropriate dtypes (numeric, string or categorical).

Note

If set to False, this will raise an exception because of a type mismatch in the OpenML dataset.

New in version 0.9.0.

cachebool, default=True

Whether to cache downloaded datasets using joblib.

data_homestr, default=None

Specify another download and cache folder for the datasets. By default, all fairlearn data is stored in ‘~/.fairlearn-data’ subfolders.

return_X_ybool, default=False

If True, returns (data.data, data.target) instead of a Bunch object.

Returns:
datasetBunch

Dictionary-like object, with the following attributes:

datandarray, shape (101766, 24)

Each row corresponding to the 24 feature values in order. If as_frame is True, data is a pandas object.

targetnumpy array of shape (101766,)

Each value represents whether readmission of the patient occurred within 30 days of the release.

feature_nameslist of length 24

Array of ordered feature names used in the dataset.

DESCRstring

Description of the Diabetes 130-Hospitals dataset.

categoriesdict or None

Maps each categorical feature name to a list of values, such that the value encoded as i is ith in the list. If as_frame is True, this is None.

framepandas DataFrame

Only present when as_frame is True. DataFrame with data and target.

(data, target)tuple if return_X_y is True

Notes

Our API largely follows the API of sklearn.datasets.fetch_openml().

References