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

Removing the warning that appears in the Graph Transformer example (#6061)

Browse Source
This commit is contained in:
Andrei Ivanov
2023-08-02 22:22:56 -07:00
committed by GitHub
parent ec7137dddf
commit 83437e67e0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/sparse/graph_transformer.py
View File

@@ -181,7 +181,7 @@ if __name__ == "__main__":
evaluator = Evaluator("ogbg-molhiv")
# laplacian positional encoding
for g, _ in tqdm(dataset, desc="Computing Laplacian PE"):
g.ndata["PE"] = dgl.laplacian_pe(g, k=pos_enc_size, padding=True)
g.ndata["PE"] = dgl.lap_pe(g, k=pos_enc_size, padding=True)
# Create model.
out_size = dataset.num_tasks