Python datasets.PPI屬性代碼示例

本文整理匯總了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

開發者ID:graph-star-team，項目名稱:graph_star，代碼行數:13，代碼來源:run_ppi.py

注：本文中的torch_geometric.datasets.PPI屬性示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台，相關代碼片段篩選自各路編程大神貢獻的開源項目，源碼版權歸原作者所有，傳播和使用請參考對應項目的License；未經允許，請勿轉載。