如何使用自定义图形数据创建pytorch几何数据集？

Question

我是 PyTorch 几何的新手，并尝试使用 SEAL_OGB 执行链接预测任务，首先我将我的图转换为 networkx 图

G = nx.read_edgelist(
    args.dataset,
    create_using=nx.DiGraph(),
    nodetype=int,
)
data = from_networkx(G)

然后使用

进行数据拆分

data = train_test_split_edges(data)

这里遇到了问题，有关于这个issue的讨论，但找不到任何解决方案，需要任何帮助！！！

顺便说一句，我正在使用 amazon dataset provide by snap， 这是我的 colab code。

数据看起来像这样 。