TimeSeriesPredictor.make_future_data_frame

TimeSeriesPredictor.make_future_data_frame(data: TimeSeriesDataFrame | DataFrame | Path | str) DataFrame[源]

生成一个数据框,其中包含与预测范围相对应的 item_idtimestamp 值。

参数:

data (Union[TimeSeriesDataFrame, pd.DataFrame, Path, str]) – 历史时间序列数据。

返回:

forecast_horizon – 数据框,其中包含与预测范围相对应的列 item_idtimestamp。对于 data 中的每个项目 ID,forecast_horizon 将包含输入数据中每个序列结束后接下来的 prediction_length 个时间步的时间戳。

返回类型:

pd.DataFrame

示例

>>> print(data)
                    target
item_id timestamp
A       2024-01-01       0
        2024-01-02       1
        2024-01-03       2
B       2024-04-07       3
        2024-04-08       4
>>> predictor = TimeSeriesPredictor(prediction_length=2, freq="D")
>>> print(predictor.make_future_data_frame(data))
  item_id  timestamp
0       A 2024-01-04
0       A 2024-01-05
1       B 2024-04-09
1       B 2024-04-10