python - 如何使用自定义图形数据创建 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)

这是遇到问题的地方，对此问题进行了讨论，但找不到任何解决方案，需要任何帮助！！！

顺便说一句，我使用的是 snap 提供的亚马逊数据集，这是我的colab 代码。

数据看起来像这样。