Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "15f6b22466d833b9c08583120bcaf953c6240351"
Unverified Commit 6a44aae1 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Cleanup example. (#6548) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent e06505b3
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 5 deletions
examples/sampling/graphbolt/quickstart/node_classification.py
View file @ 6a44aae1
...@@ -122,11 +122,6 @@ if __name__ == "__main__": ...@@ -122,11 +122,6 @@ if __name__ == "__main__":
print("Loading data...") print("Loading data...")
dataset = gb.BuiltinDataset("cora").load() dataset = gb.BuiltinDataset("cora").load()
# Uncomment to use the example cora dataset.
# dataset = gb.OnDiskDataset(
# "examples/sampling/graphbolt/quickstart/cora/"
# ).load()
in_size = dataset.feature.size("node", None, "feat")[0] in_size = dataset.feature.size("node", None, "feat")[0]
out_size = dataset.tasks[0].metadata["num_classes"] out_size = dataset.tasks[0].metadata["num_classes"]
model = GCN(in_size, out_size).to(device) model = GCN(in_size, out_size).to(device)
......
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