NeuralProphet简介
NeuralProphet是一个简单易用的时间序列预测框架,基于PyTorch构建,结合了神经网络和传统时间序列算法。它的设计灵感来自Facebook Prophet和AR-Net,旨在提供可解释的预测结果。
NeuralProphet的主要特点包括:
- 简单易学:只需几行代码即可定义、定制和评估预测模型
- 可解释性:支持可视化预测组件、模型系数等
- 人机协作:支持迭代式的人机交互建模过程
- 适用于高频长期时间序列数据
安装方法
可以直接使用pip安装NeuralProphet:
pip install neuralprophet
如果想在Jupyter notebook中使用,建议安装"live"版本:
pip install neuralprophet[live]
使用示例
以下是一个最小化示例:
from neuralprophet import NeuralProphet
# 创建模型
m = NeuralProphet()
# 拟合模型
metrics = m.fit(df)
# 预测
forecast = m.predict(df)
# 可视化结果
fig_forecast = m.plot(forecast)
fig_components = m.plot_components(forecast)
fig_model = m.plot_parameters()
预测未来数据:
m = NeuralProphet().fit(df, freq="D")
df_future = m.make_future_dataframe(df, periods=30)
forecast = m.predict(df_future)
fig_forecast = m.plot(forecast)
主要功能
NeuralProphet支持以下主要功能:
- 自回归:线性或神经网络自相关建模
- 趋势:分段线性趋势,支持自动变点检测
- 季节性:不同周期(年、日、周、时)的傅里叶项
- 滞后回归器:线性或神经网络
- 未来回归器:线性或神经网络
- 事件:国家假日和自定义重复事件
- 全局建模:支持局部、全局或"glocal"(全局+正则化局部)组件
此外还支持多时间序列建模、不确定性估计、正则化、交叉验证等功能。
NeuralProphet是一个强大而灵活的时间序列预测工具,欢迎尝试使用并参与到开源社区中来!如果您在使用过程中遇到任何问题,可以在GitHub上提issue或加入Slack社区寻求帮助.
