"git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "8086d1edde38f260ef02f0316e317ed73e019637"
Unverified Commit 241760a5 authored by Andrzej Kotłowski's avatar Andrzej Kotłowski Committed by GitHub
Browse files

[Graphbolt] Fix link_prediction example (#6397)


Co-authored-by: default avatarHongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
parent cc079e1c
...@@ -36,6 +36,7 @@ Contributors ...@@ -36,6 +36,7 @@ Contributors
* [Michal Szarmach](https://github.com/mszarma) from Intel * [Michal Szarmach](https://github.com/mszarma) from Intel
* [Izabela Mazur](https://github.com/IzabelaMazur) from Intel * [Izabela Mazur](https://github.com/IzabelaMazur) from Intel
* [Sanchit Misra](https://github.com/sanchit-misra) from Intel * [Sanchit Misra](https://github.com/sanchit-misra) from Intel
* [Andrzej Kotlowski](https://github.com/anko-intel) from Intel
* [Sheng Zha](https://github.com/szha) from AWS * [Sheng Zha](https://github.com/szha) from AWS
* [Yifei Ma](https://github.com/yifeim) from AWS * [Yifei Ma](https://github.com/yifeim) from AWS
* [Yizhi Liu](https://github.com/yzhliu) from AWS * [Yizhi Liu](https://github.com/yzhliu) from AWS
......
...@@ -169,7 +169,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True): ...@@ -169,7 +169,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True):
############################################################################ ############################################################################
# [Step-4]: # [Step-4]:
# gb.to_dgl() # datapipe.to_dgl()
# [Input]: # [Input]:
# 'datapipe': The previous datapipe object. # 'datapipe': The previous datapipe object.
# [Output]: # [Output]:
...@@ -177,7 +177,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True): ...@@ -177,7 +177,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True):
# [Role]: # [Role]:
# Convert a mini-batch to dgl-minibatch. # Convert a mini-batch to dgl-minibatch.
############################################################################ ############################################################################
datapipe = gb.to_dgl() datapipe = datapipe.to_dgl()
############################################################################ ############################################################################
# [Input]: # [Input]:
......
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