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

问题描述

我是 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

数据看起来像这样

data image

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)