本文整理汇总了Python中torch_geometric.datasets.PPI属性的典型用法代码示例。如果您正苦于以下问题：Python datasets.PPI属性的具体用法？Python datasets.PPI怎么用？Python datasets.PPI使用的例子？那么恭喜您, 这里精选的属性代码示例或许可以为您提供帮助。您也可以进一步了解该属性所在类torch_geometric.datasets的用法示例。
在下文中一共展示了datasets.PPI属性的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: load_data
# 需要导入模块: from torch_geometric import datasets [as 别名]
# 或者: from torch_geometric.datasets import PPI [as 别名]
def load_data():
path = osp.join(osp.dirname(osp.realpath(__file__)), '.', 'data', 'PPI')
train_dataset = PPI(path, split='train')
val_dataset = PPI(path, split='test')
test_dataset = PPI(path, split='test')
train_loader = DataLoader(train_dataset, batch_size=1, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=2, shuffle=False)
test_loader = DataLoader(test_dataset, batch_size=2, shuffle=False)
return train_dataset, val_dataset, test_dataset, train_loader, val_loader, test_loader