From 3719ada7d809258b843ee1448a7ca4676b75eae6 Mon Sep 17 00:00:00 2001 From: Matthias Fey Date: Mon, 23 Mar 2020 09:53:55 +0100 Subject: [PATCH] Update main_pyg.py --- examples/graphproppred/ppi/main_pyg.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/graphproppred/ppi/main_pyg.py b/examples/graphproppred/ppi/main_pyg.py index 70c13a3d..0f1ecfba 100644 --- a/examples/graphproppred/ppi/main_pyg.py +++ b/examples/graphproppred/ppi/main_pyg.py @@ -11,10 +11,7 @@ ### importing OGB -### for loading dataset -from ogb.graphproppred.dataset_pyg import PygGraphPropPredDataset -### for evaluation -from ogb.graphproppred import Evaluator +from ogb.graphproppred import PygGraphPropPredDataset, Evaluator multicls_criterion = torch.nn.CrossEntropyLoss() @@ -149,4 +146,4 @@ def main(): if __name__ == "__main__": - main() \ No newline at end of file + main()