#17 · 主分类: 时间序列机器学习

pytorch_geometric_temporal

deep-learning gcn gnn graph-convolution graph-convolutional-networks graph-embedding graph-neural-networks network-embedding network-science node-embedding pytorch spatial-analysis spatial-data spatio-temporal-analysis spatio-temporal-data spatiotemporal temporal-data temporal-graphs temporal-networks

PyTorch Geometric Temporal:基于神经机器学习模型的时空信号处理(CIKM 2021)

项目最后更新:05/30/26

GitHub Stars

3.0K

Forks数量

403

贡献者数量

40

许可证

MIT

收录理由

多数时间序列工具会忽略数据背后的图结构,而图算法库又常常不关心时间维度。PyTorch Geometric Temporal正好补上这块空白:它是PyTorch Geometric的动态扩展,让你能在连续图快照上训练图神经网络,实现来自多篇已发表论文的方法,覆盖交通流量、疫情传播、风电场出力等时空回归场景。库本身也把那些不起眼但关键的细节处理好了——静态或动态图的快照迭代器、训练测试切分,以及METR-LA、PEMS-BAY等基准数据集,方便你和论文中的结果对照。做研究的人可以站在大量近期的模型基础上,工程上也能省下自己写批处理和切分的功夫,快速搭出可运行的验证。如果你已经用PyTorch且习惯以图的方式思考问题,用它试一个时空图的想法会很顺手。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目