1. PyTorch Geometric Temporal 介绍
PyTorch Geometric Temporal 是 PyTorch Geometric 的一个时间(动态)扩展库。
该图书馆包括各种动态和时间几何深度学习,嵌入和时空回归方法从各种已发表的研究论文。此外,它还附带了一个易于使用的数据集加载器、列车测试分离器和用于动态和时态图的时态快速迭代器。该框架自然地提供了 GPU 支持。它还提供了大量的基准数据集,包括人口统计学预测、共享经济、能源生产和网络流量管理等领域。最后,您还可以创建自己的数据集。
该软件包与 Pytorch Lightning 接口良好,允许在 cpu 上进行培训,单个和多个 gpu 即开即用。看看这个使用 PyTorch Geometric Temporal 和 PyTorch lightning 的介绍性示例。
2. Github 地址:GitHub - benedekrozemberczki/pytorch_geometric_temporal: PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
下载ZIP: