存储多个 XGBoost 模型

Question

我有一个时间序列数据集。我想在时间序列中每天训练一个 XGB 模型。我想使用 for 循环，但我不确定如何正确存储所有模型并在以后调用它们。这是我到目前为止所拥有的：

for date in range(minDate, maxDate):
   model[date] = xgb.train(params, data)

但我不确定 model[date] 应该是什么？它可以是一个 numpy 数组吗？

Answer 1

您可以使用以日期为键的字典：

model = {}
for date in range(minDate, maxDate):
   model[date] = xgb.train(params, data)

然后您可以为给定日期调用任何模型随机日期，假设给定一些输入来预测输出

model[RandomDate].predict(inputs)

Storing multiple XGBoost models