使用单行代码在时间序列数据集上自动构建 ARIMA、SARIMAX、VAR、FB Prophet 和 XGBoost 模型。现在用 Dask 更新以处理数百万行。
本文使用了从 Kaggle 下载的 2006 年 1 月至 2018 年 1 月的亚马逊股票价格[2]数据集。该库仅提供训练时间序列预测模型。数据集应该有一个时间或日期格式列。
初始化 Auto-TS 模型对象,并拟合训练数据:
model = auto_timeseries(
forecast_period=219,
score_type='rmse',
time_interval='D',
model_type='best')
model.fit(traindata= train_df,
ts_column="Date",
target="Close")