THUDM / CogDL

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
https://cogdl.ai
MIT License
1.73k stars 315 forks source link

制作自己的数据集 #443

Open oxon-612 opened 5 months ago

oxon-612 commented 5 months ago

❓ Questions & Help

您好,我想借助cogDL在我的应用领域开展预测,因此需要将自己的数据集导入模型,请问如何制作自己的数据集,有没有说明文档,因为目前发现可以引入GNN模型的数据集结构各异。 谢谢您的帮助。

oxon-612 commented 5 months ago

特别的,我想做的数据集是包含时间信息的,但是在自定义数据集的教程中看到的图结构目前还不带时间信息,请问如何考虑加入时间序列信息呢,以及我的边不是0/1是否连接,而是具体的连接量,请问这一点如何解决,感谢您的帮助。

cenyk1230 commented 5 months ago

自定义数据集的构建可以看这个用法:https://github.com/THUDM/CogDL/blob/master/examples/legacy/custom_dataset.py