- Python3.7
- tensorflow-gpu>=2.0
- Numpy==1.19.5
- CUDA 11.0+
- Pandas==1.2.4
- Scipy==1.6.3
- Pyhht==0.1.0
- Matplotlib==3.2.2
由9个单元组成,其中多头注意力LSTM单元3个,LSTM单元6个,两种单元结构如下:
- 模型由9个单元组成,共两种类型:LSTM单元与多头注意力LSTM单元
- 根据经验模态分解,将输入数据拆分为9组,并按pearson相关系数排序
- 模型误差由9组imf分量共同决定,提高泛化性
- 数据路径、训练参数均位于config.py
链接:https://pan.baidu.com/s/1PYSmTSc7ucK3nYCpyEeb5A 提取码:sets 下载解压后放置于config.py中设置的路径即可。
1.默认设置某气候数据为训练样本 2.运行train.py