Skip to content

GitLab

"tutorials/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "2a92dfca306562e403b7310046c2dfeb506b23ed"
Unverified Commit e4bb6abc authored by yxy235's avatar yxy235 Committed by GitHub
Browse files

[Misc] Replace API to get feature size in example `link prediction`. (#6458) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
parent c5ea35dc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/sampling/graphbolt/link_prediction.py
View file @ e4bb6abc
...@@ -348,7 +348,7 @@ def main(args): ...@@ -348,7 +348,7 @@ def main(args):
valid_set = dataset.tasks[0].validation_set valid_set = dataset.tasks[0].validation_set
args.fanout = list(map(int, args.fanout.split(","))) args.fanout = list(map(int, args.fanout.split(",")))
in_size = 128 in_size = features.size("node", None, "feat")[0]
hidden_channels = 256 hidden_channels = 256
model = SAGE(in_size, hidden_channels) model = SAGE(in_size, hidden_channels)
......
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