[GNN] pyogd examples

Author

김보람

Published

January 18, 2024


# import torch_geometric.transforms as T
# from torch_geometric.datasets import Planetoid

# data = Planetoid('./data/Cora', 'Cora', transform=T.NormalizeFeatures())[0]
# import torch
# from pygod.generator import gen_contextual_outlier, gen_structural_outlier

data


# data, ya = gen_contextual_outlier(data, n=100, k=50)
# data, ys = gen_structural_outlier(data, m=10, n=10)
# data.y = torch.logical_or(ys, ya).long()
# data
# from pygod.utils import load_data

# data = load_data('inj_cora')
# data.y = data.y.bool()
# data

Initialization

# from pygod.detector import DOMINANT

# detector = DOMINANT(hid_dim=64, num_layers=4, epoch=100)
# detector.fit(data)