时空序列预测PyTorch模型集
此数据集包含基于 PyTorch 的时空序列预测模型。模型结构代码储存在 models 文件夹中,详细展示了论文中公式和图表。模型假设输入张量的形状为 (batch, sequence, channel, height, width)。
此外,还提供了以下文件夹:
- util:包含进行 patch 分割的大数据处理方法
- TrainingTemplate 和 TestingTemplate:用于训练和测试的模板类
- content_tree:包含生成目录树的方法
下载地址
用户评论