Unverified Commit 3dd54d5c authored by Xin Zhang's avatar Xin Zhang Committed by GitHub
Browse files

fix friendster dataset uasge (#4009)


Co-authored-by: default avatarRhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
parent de5543e3
......@@ -91,7 +91,8 @@ def get_graph(name, format = None):
g_list, _ = dgl.load_graphs(bin_path)
g = g_list[0]
else:
g = get_friendster().formats(format)
# the original node IDs of friendster are not consecutive, so we compact it
g = dgl.compact_graphs(get_friendster()).formats(format)
dgl.save_graphs(bin_path, [g])
elif name == "reddit":
bin_path = "/tmp/dataset/reddit/reddit_{}.bin".format(format)
......
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