NeuralProphet入门指南 - 简单易用的时间序列预测框架

neural_prophet

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)