python - 使用 facebook 先知在具有多个时间序列的数据框中进行时间序列预测

Question

我有以下数据框：

             fid         via
2015-01-18  id_22207  0.275056
2015-01-30  id_22207  0.306961
2015-02-23  id_22207  0.285065
2015-02-24  id_22207  0.337570
2015-02-27  id_22207  0.311612
2015-01-18  id_22208  0.371765
2015-01-20  id_22208  0.405391
2015-02-11  id_22208  0.354052
2015-02-24  id_22208  0.421126
2015-03-15  id_22208  0.454406

我想使用这个数据框使用 facebook 的prophet库进行时间序列预测。该库中是否有使用此数据框的方法？棘手的部分是我有多个 fid 值，并且对于每个 fid，我在via列中都有多个日期的数据。我想为下面的via列做预测。数据框：

2015-03-18  id_22209
2015-03-20  id_22209
2015-03-21  id_22209
2015-03-24  id_22209
2015-03-25  id_22209

Answer 1

目前，prophet不支持多元时间序列预测或 VAR。您最好的选择是在基于fid列拆分数据框后循环创建预测。

library(tidyverse)
library(prophet)

lapply(split(df, f= df$fid), function(x) {
    # Prophet expects columns to be ds, y
    x <- x %>% rename(y = via, ds = date)

    # Create prophet forecasts
    # ...
})

---

编辑

没有注意到这个问题被标记为 Python。

unique_fid = df['fid'].unique()
for fid in unique_fid:
    temp_df = df.loc[df['fid'] == fid,['date', 'via']]
    # Prophet expects ds and y as columns
    temp_df.columns = ['ds', 'y']
    # Create prophet forecasts
    # ...