Unverified Commit 609a49a9 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[Doc] fix example in dataset adapter docstring (#3772)

parent 2f4146a4
......@@ -228,10 +228,10 @@ class AsLinkPredDataset(DGLDataset):
>>> ds = dgl.data.CoraGraphDataset()
>>> print(ds)
Dataset("cora_v2", num_graphs=1, save_path=...)
>>> new_ds = dgl.data.AsNodePredDataset(ds, [0.8, 0.1, 0.1])
>>> new_ds = dgl.data.AsLinkPredDataset(ds, [0.8, 0.1, 0.1])
>>> print(new_ds)
Dataset("cora_v2-as-linkpred", num_graphs=1, save_path=/home/ubuntu/.dgl/cora_v2-as-linkpred)
>>> print(hasattr(new_ds, "get_test_edges"))
>>> print(hasattr(new_ds, "test_edges"))
True
"""
......
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