Datasets

aif360.datasets

Base classes

datasets.Dataset(**kwargs)

Abstract base class for datasets.

datasets.StructuredDataset(df, label_names, ...)

Base class for all structured datasets.

datasets.BinaryLabelDataset([...])

Base class for all structured datasets with binary labels.

datasets.StandardDataset(df, label_name, ...)

Base class for every BinaryLabelDataset provided out of the box by aif360.

datasets.RegressionDataset(df, dep_var_name, ...)

Base class for regression datasets.

Common datasets

datasets.AdultDataset([label_name, ...])

Adult Census Income Dataset.

datasets.BankDataset([label_name, ...])

Bank marketing Dataset.

datasets.CompasDataset([label_name, ...])

ProPublica COMPAS Dataset.

datasets.GermanDataset([label_name, ...])

German credit Dataset.

datasets.LawSchoolGPADataset([dep_var_name, ...])

Law School GPA dataset.

datasets.MEPSDataset19([label_name, ...])

MEPS Dataset.

datasets.MEPSDataset20([label_name, ...])

MEPS Dataset.

datasets.MEPSDataset21([label_name, ...])

MEPS Dataset.