dig.fairgraph.dataset¶

`NBA`	NBA is an NBA on court performance dataset along salary, social engagement etc.
`POKEC`	Pokec is a social network dataset.

class NBA(data_path='https://github.com/divelab/DIG_storage/raw/main/fairgraph/datasets/nba/', root='./dataset/nba')[source]¶

NBA is an NBA on court performance dataset along salary, social engagement etc.

Parameters

data_path (str, optional) – The url where the dataset is found, defaults to https://github.com/divelab/DIG_storage/raw/main/fairgraph/datasets/nba/
root (str, optional) – The path to root directory where the dataset is saved, defaults to /dataset/nba

class POKEC(data_path='https://github.com/divelab/DIG_storage/raw/main/fairgraph/datasets/pockec/', root='./dataset/pokec', dataset_sample='pokec_z')[source]¶

Pokec is a social network dataset. Two different datasets (namely pokec_z and pokec_n) are sampled from the original Pokec dataset.

Parameters

data_path (str, optional) – The url where the dataset is found, defaults to https://github.com/divelab/DIG_storage/raw/main/fairgraph/datasets/pockec/
root (str, optional) – The path to root directory where the dataset is saved, defaults to /dataset/pokec
dataset_sample (str, optional) – The sample (should be one of pokec_z or pokec_n) to be used in choosing the POKEC dataset. Defaults to pokec_z

Raises

Exception When invalid dataset_sample is provided.