Unverified Commit 09c33b9f authored by xiangyuzhi's avatar xiangyuzhi Committed by GitHub
Browse files

[Sparse] Fix graph_transformer example (#6471)

parent 101d2ae9
...@@ -3684,12 +3684,13 @@ def lap_pe(g, k, padding=False, return_eigval=False): ...@@ -3684,12 +3684,13 @@ def lap_pe(g, k, padding=False, return_eigval=False):
if k + 1 < n - 1: if k + 1 < n - 1:
# Use scipy if k + 1 < n - 1 for memory efficiency. # Use scipy if k + 1 < n - 1 for memory efficiency.
EigVal, EigVec = scipy.sparse.linalg.eigs( EigVal, EigVec = scipy.sparse.linalg.eigs(
L, k=k + 1, which="SR", tol=1e-2 L, k=k + 1, which="SR", ncv=4 * k, tol=1e-2
) )
max_freqs = k
topk_indices = EigVal.argsort()[1:] topk_indices = EigVal.argsort()[1:]
# Since scipy may return complex value, to avoid crashing in NN code, # Since scipy may return complex value, to avoid crashing in NN code,
# convert them to real number. # convert them to real number.
topk_eigvals = EigVal[topk_indices].real topk_EigVal = EigVal[topk_indices].real
topk_EigVec = EigVec[:, topk_indices].real topk_EigVec = EigVec[:, topk_indices].real
else: else:
# Fallback to numpy since scipy.sparse do not support this case. # Fallback to numpy since scipy.sparse do not support this case.
......
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