python - 在 Python 中计算 Prophet 模型的 RMSE

Question

我是学习 python 的新手，我正在创建一个时间序列模型，该模型使用 Prophet 模型预测每个月的总销售额。我需要帮助获取模型的 RMSE！谢谢！

from fbprophet import Prophet
#Creating Dataframe#
proph = train.groupby(['date_block_num'])[ 'item_cnt_day'].sum()
proph.index=pd.date_range(start='2013-01-01', end='2015-10-01', freq='MS')
proph = proph.to_frame().reset_index()
proph.columns = ['ds', 'y']
proph.head()
#Modelling#
model=Prophet(yearly_seasonality=True)
model.fit(proph)
model.head()
#Making future predictions
future_data = model.make_future_dataframe(periods=1, freq='MS')
forecast = model.predict(future_data)
forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail()

Answer 1

我不知道它是否仍然相关。您将需要准备一个保存实际值的 DataFrame，让我们调用它df_actual。然后以下将为您计算 RMSE：

se = np.square(forecast.loc[:, 'yhat'] - df_actual)
mse = np.mean(se)
rmse = np.sqrt(mse)

希望这可以帮助。