indexError：只有整数，切片（`：`），省略（`...

Question

我想实现先知算法并在模型上使用我的数据集。 在拟合过程之后，我在预测过程中遇到了以下错误。 我该如何解决这个问题？

import pandas as pd
import pystan
from prophet import Prophet
df_prophet = df[['date', 'rate']]
train_df = df_prophet[:-5]
train_df.columns = ['ds', 'y']
train_df['ds']= to_datetime(train_df['ds'])
model = Prophet()
model.fit(train_df)
test_df = df_prophet[-5:][['date']]
test_list = to_datetime(test_df.date).values
forecast = model.predict(test_list)

----＆gt; 11预测= model.predict（test_list）

indexError：只有整数，切片（：），省略（...）， numpy.newaxis（无）和整数或布尔数组是有效的索引

Answer 1

这是一个很好的尝试。您只需要进行一些调整。

1. 通过传递 to_datetime(test_df.date).values ，您正在创建一个 numpy 数组而不是数据帧。 Prophet 需要一个数据帧。
2. Prophet 模型和预测器需要标记为 ds 和 y 的列，并且您在拆分数据帧后更改列名称，因此您的测试部分没有重命名列。
3. 您不需要导入 pystan，因为 Prophet 模块已经构建在它的基础上。

试试这个：

import pandas as pd
from prophet import Prophet

df_prophet = df[['Date', 'Volume']]
df_prophet.columns = ['ds', 'y']
train_df = df_prophet[:-5]
train_df['ds']= pd.to_datetime(train_df['ds'])
model = Prophet()
model.fit(train_df)
test_df = df_prophet[-5:][['ds']]
forecast = model.predict(test_df)
forecast