本文包含什么?
- TFT+机器学习融合
- 完整的在线运行的代码环境(免配置环境)
- 代码介绍
- 运行有问题? csdn上后台随时售后.
项目说明
本文主要实现用谷歌的论文Temporal Fusion Transformers for Interpretable Multi-horizon Time Series Forecasting(TFT)来做时间序列的预测.
模型整体结构如图所示:
本文评价指标主要是MAE
模型 | MAE |
---|---|
TFT | 13.351490879058838 |
xgb | 12.108228302001953 |
xgb+TFT | 10.959671516418457 |
代码运行介绍
代码地址在文末
点击编辑后注意选择P100的GPU以及打开internet的开关:
首先将准备的代码以及数据集拷贝到当前路径:
%cp