使用 Facebook Prophet 一次预测多个变量

Question

我是 Python 和 Facebook Prophet 的新手，所以这可能很简单，但我一直无法在线找到答案。

我有一个 7 列的 csv 文件。一列包含每日增量的日期戳 ('ds') 列，其他 6 列（'y1'、'y2'、'y3' 等）包含 6 个值对齐的变量带有日期戳。

我不想创建六个不同的两列 csv 文件和运行ning Prophet 六个不同的时间（一次只预测一个变量），而是想找到一种方法来预测所有六个变量一次。这是我正在尝试的：

df = pd.read.csv('example_file.csv')
cols = ['y1','y2','y3','y4','y5','y6']
results = []
for col in cols:
    subdf = df[['ds', col]].dropna()
    m = Prophet()
    m.fit(subdf)
    result = m.predict(m.make.future.dataframe(periods = 90))
    results.append(result)
df.predict = pd.concat(results, axis=1)
df.predict.to_csv('example_file.csv')

当我运行它时，出现以下错误：

ValueError: Dataframe must have columns 'ds' and 'y' with the dates and values respectively.

任何 insight/help 将不胜感激。谢谢！

Answer 1

抱歉，我想发表评论，但我还没有足够的声誉。请在循环中重命名您的列

subdf = subdf.rename(columns={'ds':'ds', col:'y'})

Prophet 强加了输入列命名为 ds（时间列）和 y（指标列）的严格条件。

使用 Facebook Prophet 一次预测多个变量

Predicting multiple variables at once with Facebook Prophet

python

pandas

facebook-prophet