Open Graph Benchmark 数据集加载

以图分类数据集 molhiv 为例：

from ogb.graphproppred import PygGraphPropPredDataset
# from ogb.nodeproppred import PygNodePropPredDataset
# import torch_geometric

from torch_geometric.data import DataLoader

if __name__ == '__main__':
    dataset = PygGraphPropPredDataset(name="ogbg-molhiv", root='/home/yingpan/disk1/lzy_data/dataset/')
    split_idx = dataset.get_idx_split()
    train_loader = DataLoader(dataset[split_idx["train"]], batch_size=32, shuffle=True)
    valid_loader = DataLoader(dataset[split_idx["valid"]], batch_size=32, shuffle=False)
    test_loader = DataLoader(dataset[split_idx["test"]], batch_size=32, shuffle=False)

最后

以上就是悲凉老师为你收集整理的Open Graph Benchmark 数据集加载的全部内容，希望文章能够帮你解决Open Graph Benchmark 数据集加载所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。