时间序列预测 | Python实现Attention-Transformer时间序列预测
基本介绍
Python实现Attention-Transformer时间序列预测(TSAT model)
模型结构
main.py
:含训练集合测试集的模型主程序TSAT.py
: 时间注意力模块utils.py
: 函数与参数设定dataset_TSAT_ETTm1_48.py
: 从ETTm1数据集生成图
参数设置可以在 utils.py 中找到
l_backcast
: 回溯的长度d_edge
: 使用的 IMF 数量d_model
:时间嵌入维度N
: Self_Attention_Block 的数量h
: 多头注意力中的头数N_dense
:顺序前馈层中的线性层数n_output
: 输出数(预测长度