pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Graphbolt] Fix link_prediction example (#6397)

Browse Source 
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
This commit is contained in:
Andrzej Kotłowski
2023-10-09 07:18:20 +02:00
committed by GitHub
parent cc079e1c44
commit 241760a5ef
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
CONTRIBUTORS.md
View File

@@ -36,6 +36,7 @@ Contributors
* [Michal Szarmach](https://github.com/mszarma) from Intel
* [Izabela Mazur](https://github.com/IzabelaMazur) 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
* [Yifei Ma](https://github.com/yifeim) from AWS
* [Yizhi Liu](https://github.com/yzhliu) from AWS

4
examples/sampling/graphbolt/link_prediction.py
View File

@@ -169,7 +169,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True):
############################################################################
# [Step-4]:
# gb.to_dgl()
# datapipe.to_dgl()
# [Input]:
# 'datapipe': The previous datapipe object.
# [Output]:
@@ -177,7 +177,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True):
# [Role]:
# Convert a mini-batch to dgl-minibatch.
############################################################################
datapipe = gb.to_dgl()
datapipe = datapipe.to_dgl()
############################################################################
# [Input]: