n-beats: Keras与Pytorch的N-BEATS实现
简介
N-BEATS是一种用于可解释时间序列预测的神经网络模型,由Philippe Remy和Jean-Sebastien Dhr提出。本仓库提供了N-BEATS在Keras和Pytorch两种框架下的实现代码。通过该模型,用户可以有效地对时间序列数据进行预测。
安装与使用
环境准备
请确保您在virtualenv环境中,并已安装Python 3。
安装Keras版本
通过PyPI安装:
pip install nbeats-keras
从源代码安装:
make install-keras
安装Pytorch版本
通过PyPI安装:
pip install nbeats-pytorch
从源代码安装:
make install-pytorch
GPU运行
如需在GPU上运行,请确保已正确配置CUDA环境。
注意事项
- 确保在virtualenv环境中操作。
- 本仓库提供的代码仅供参考与学习,如有问题,请自行承担相应责任。
感谢您的关注与使用!