我正在使用XGBoostRegressor,设置两个参数num_early_stopping_rounds
和maximize_evaluation_metrics
。当尽早停止时,有什么办法可以使迭代次数增加?
如source code of XGBoost.java
所示:
if (earlyStoppingRounds > 0) {
boolean onTrack = judgeIfTrainingOnTrack(params, earlyStoppingRounds, metrics, iter);
if (!onTrack) {
String reversedDirection = getReversedDirection(params);
Rabit.trackerPrint(String.format(
"early stopping after %d %s rounds", earlyStoppingRounds, reversedDirection));
break;
}
}
这里的iter
可能是训练模型的最佳方法,但是我不知道如何提取此变量。
非常感谢您。