xgboost実行時間を制限する方法は？

Question

xgboostの実行時間を制限する方法（たとえば、クラスタに1時間の計算予算があります）？ラウンド数でxgboostを制限することも可能ですか？

Answer 1

で使用されているxgboost.train関数を直接呼び出すことは可能です。この関数には、実行時間を確認し、必要に応じてEarlyStopExceptionを呼び出すために使用できるパラメータnum_boost_round（昇圧ラウンド数を制限する）とcallbacksパラメータがあります。もちろん、反復の途中で停止することはできませんが、time_limit - delta分です。

残念ながら、XGBModel（およびそれに対応XGBRegressorとXGBClassifier）APIを使用すると、低レベルのAPIにコードをリファクタリングする必要がありますので、コールバックを通過することはできません：

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