使用Spark xgboost4j提前停止时如何获取迭代?

时间:2018-11-22 22:00:27

标签: scala apache-spark xgboost

我正在使用XGBoostRegressor,设置两个参数num_early_stopping_roundsmaximize_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可能是训练模型的最佳方法,但是我不知道如何提取此变量。

非常感谢您。

0 个答案:

没有答案