Skip to content

Instantly share code, notes, and snippets.

View hongxuenong's full-sized avatar

Shannon Hong hongxuenong

  • Shopee
  • Singapore
View GitHub Profile
from math import sqrt
import torch
from tqdm import tqdm
import matplotlib.pyplot as plt
import networkx as nx
from torch_geometric.nn import MessagePassing
from torch_geometric.data import Data
from torch_geometric.utils import k_hop_subgraph, to_networkx
import numpy as np