-
KerasPytorch中对N-BEATS模型的实现
资源介绍
N-BEATS:神经基础扩展分析,用于可解释的时间序列预测(Keras,Pytorch)
链接到[]。 作者:Philippe Remy和Jean-Sebastien Dhr
训练开始时的N-Beats
相信我,再走几步,绿色曲线(预测值)就会与地面真实情况完全匹配:-)
安装
确保您在virtualenv中(推荐)并安装了python3。
从PyPI
pip install nbeats-keras : pip install nbeats-keras 。
安装Pytorch: pip install nbeats-pytorch 。
从来源
安装基于MakeFile。
使用Keras安装N-Beats的命令: make install-keras
使用Pytorch安装N-Beats的命令: make install-pytorch
在GPU上运行
要强制使用GPU(带