Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "9e6bdbabcf2a39563d69dad81d8fddef5800ae3f"
Unverified Commit d1124b7b authored by 彭齐荣's avatar 彭齐荣 Committed by GitHub
Browse files

[Example][BugFix] Fix Lightning GraphSAGE examples (dmlc#4046) (#4053)

parent 3c129ad7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/pytorch/graphsage/lightning/node_classification.py
View file @ d1124b7b
...@@ -116,6 +116,7 @@ if __name__ == '__main__': ...@@ -116,6 +116,7 @@ if __name__ == '__main__':
dataset = DglNodePropPredDataset('ogbn-products') dataset = DglNodePropPredDataset('ogbn-products')
graph, labels = dataset[0] graph, labels = dataset[0]
graph.ndata['label'] = labels.squeeze() graph.ndata['label'] = labels.squeeze()
graph.create_formats_()
split_idx = dataset.get_idx_split() split_idx = dataset.get_idx_split()
train_idx, val_idx, test_idx = split_idx['train'], split_idx['valid'], split_idx['test'] train_idx, val_idx, test_idx = split_idx['train'], split_idx['valid'], split_idx['test']
datamodule = DataModule(graph, train_idx, val_idx, [15, 10, 5], 1024, dataset.num_classes) datamodule = DataModule(graph, train_idx, val_idx, [15, 10, 5], 1024, dataset.num_classes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment