Python 如何限制xgboost执行时间？

如何限制xgboost执行时间（例如，我在集群中有一个1小时的“计算预算”？也可以限制xgboost的轮数吗？

如果您直接调用函数，这是XGBModel使用的函数。请将安装在发动机罩下。此函数有一个

num\u boost\u round

参数（用于限制boosting round的数量）和

callbacks

参数，可用于检查运行时间并在需要时引发

earlystopeexception

。当然，在迭代过程中不能停止，但在<代码> Time-限制- delta /代码>分钟。

不幸的是，

XGBModel

（以及相应的

XGBRegressor

和

XGBClassifier

）API不允许通过回调，因此您必须将代码重构为低级API：

params = {'eval_metric': True, ...}
trainDmatrix = DMatrix(X, label=y)
train(params, trainDmatrix, num_boost_round=10, callbacks=[callback])

---

为了澄清，您希望根据执行这些任务所需的时间来设置轮数？@dataprincess否，这是单独的问题。