#17 · 主分类: 时间序列机器学习
pytorch_geometric_temporal
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且习惯以图的方式思考问题,用它试一个时空图的想法会很顺手。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。