1. 首页
  2. 人工智能
  3. 机器学习
  4. 时空序列预测PyTorch模型集

时空序列预测PyTorch模型集

上传者: 2024-05-02 12:00:16上传 ZIP文件 53.55KB 热度 8次

此数据集包含基于 PyTorch 的时空序列预测模型。模型结构代码储存在 models 文件夹中,详细展示了论文中公式和图表。模型假设输入张量的形状为 (batch, sequence, channel, height, width)

此外,还提供了以下文件夹:

- util:包含进行 patch 分割的大数据处理方法

- TrainingTemplate 和 TestingTemplate:用于训练和测试的模板类

- content_tree:包含生成目录树的方法

用户评论