NBeats公司
nbeats-pytorch的Python项目详细描述
N-拍:可解释时间序列预测的神经基展开分析
- 在Pytorch中的实现
- 在Keras中的实现由@eljdos
- https://arxiv.org/abs/1905.10437
相信我,再多走几步,绿色曲线(预测)完全符合基本事实:-)
安装
从PyPI
安装Keras:pip install nbeats-keras
。在
安装Pytorch:pip install nbeats-pytorch
。在
来源
安装基于生成文件。确保您使用的是virtualenv并安装了python3。在
使用Keras安装N-Beats的命令:make install-keras
使用Pytorch安装N-Beats的命令:make install-pytorch
示例
Jupyter笔记本:NBeats.ipynb:make run-jupyter
。在
型号
Pythorch和Keras有相同的模型参数:
classNBeatsNet:def__init__(self,stack_types=[TREND_BLOCK,SEASONALITY_BLOCK],nb_blocks_per_stack=3,forecast_length=2,backcast_length=10,thetas_dim=[2,8],share_weights_in_stack=False,hidden_layer_units=128):pass
在这个模型中:
^{pr2}$- 项目
标签: