Utils#
- class pygod.utils.load_data(name, cache_dir=None)[source]#
Data loading function. See data repository for supported datasets. For injected/generated datasets, the labels meanings are as follows.
0: inlier
1: contextual outlier only
2: structural outlier only
3: both contextual outlier and structural outlier
- Parameters:
- Returns:
data – The outlier dataset.
- Return type:
torch_geometric.data.Data
Examples
>>> from pygod.utils import load_data >>> data = load_data(name='weibo') # in PyG format >>> y = data.y.bool() # binary labels (inlier/outlier) >>> yc = data.y >> 0 & 1 # contextual outliers >>> ys = data.y >> 1 & 1 # structural outliers