以图分类数据集 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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复