Prophet Python ValueError:数据帧中缺少Regressor

Question

我正在尝试使用Python的Prophet包的最新(第二个) 0.3版本。

我的模型应该包括一个外生回归器，但我收到一个ValueError，指出数据帧中缺少真正存在的回归器。这是一个bug，还是我做错了什么？

#Random Dataset Preparation

import random
random.seed(a=1)

df = pandas.DataFrame(data = None, columns = ['ds', 'y', 'ex'], index = range(50))
datelist = pandas.date_range(pandas.datetime.today(), periods = 50).tolist()

y = numpy.random.normal(0, 1, 50)
ex = numpy.random.normal(0, 2, 50)

df['ds'] = datelist
df['y'] = y
df['ex'] = ex

#Model
prophet_model = Prophet(seasonality_prior_scale = 0.1)
Prophet.add_regressor(prophet_model, 'ex')
prophet_model.fit(df)
prophet_forecast_step = prophet_model.make_future_dataframe(periods=1)

#Result-df
prophet_x_df = pandas.DataFrame(data=None, columns=['Date_x', 'Res'], index = range(int(len(y))))

#Error
prophet_x_df.iloc[0,1] = prophet_model.predict(prophet_forecast_step).iloc[0,0] 

Answer 1

您需要首先创建一个具有回归量值的列，该列需要出现在拟合和预测数据帧中。

Refer prophet docs

Answer 2

make_future_dataframe仅生成包含ds列的数据帧。您需要将'ex‘列添加到prophet_forecast_step数据框中，以便将其用作回归器