holisticai.datasets.load_hai_datasets#

holisticai.datasets.load_hai_datasets(dataset_name, data_home=None)[source]#

Generic function to load datasets from holisticai datasets repository.

Available datasets: - adult - law_school - student - lastfm - us_crime - clinical_records - acsincome - acspublic - compass_two_year_recid - compass_is_recid - german_credit - bank_marketing - census_kdd - diabetes - mw_small - mw_medium

Parameters

namestr

Name of the dataset.

versionint

Version of the dataset.

data_homestr, optional

The directory to which the data is downloaded. If None, we download to the default data home directory.

Returns

datapd.DataFrame

The dataset from holisticai datasets.

References