是的,auto arima预测函数会返回一个包含预测值的时间序列。以下是一个代码示例:
# 导入需要的库
import pandas as pd
import numpy as np
from pmdarima.arima import auto_arima
# 创建时间序列
date_rng = pd.date_range(start='1/1/2020', end='1/31/2020', freq='D')
values = [np.random.randint(1, 30) for i in range(31)]
ts = pd.Series(values, index=date_rng)
# 构建auto arima模型并进行预测
arima_model = auto_arima(ts, seasonal=False, suppress_warnings=True)
forecast = arima_model.predict(n_periods=7)
# 打印预测结果
print(forecast)
在此示例中,auto arima预测函数返回一个包含7个预测值的时间序列。