From 5f8c56dcd77ca23e243f9ad7c657ba6c60dc2d91 Mon Sep 17 00:00:00 2001 From: Adam Valenta Date: Fri, 15 Sep 2023 19:07:36 +0200 Subject: [PATCH] add log --- h2o-algos/src/main/java/hex/adaboost/AdaBoost.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/h2o-algos/src/main/java/hex/adaboost/AdaBoost.java b/h2o-algos/src/main/java/hex/adaboost/AdaBoost.java index 897aabe0cfa7..7a5c08f29c5e 100644 --- a/h2o-algos/src/main/java/hex/adaboost/AdaBoost.java +++ b/h2o-algos/src/main/java/hex/adaboost/AdaBoost.java @@ -12,6 +12,7 @@ import water.exceptions.H2OModelBuilderIllegalArgumentException; import water.fvec.Frame; import water.fvec.Vec; +import water.util.Timer; import water.util.TwoDimTable; import java.util.ArrayList; @@ -110,6 +111,7 @@ private void buildAdaboost() { } for (int n = 0; n < _parms._n_estimators; n++) { + Timer timer = new Timer(); ModelBuilder job = chooseWeakLearner(_trainWithWeights); job._parms._seed += n; Model model = (Model) job.trainModel().get(); @@ -128,6 +130,8 @@ private void buildAdaboost() { updateWeightsTask.doAll(_trainWithWeights.vec(_weightsName), _trainWithWeights.vec(_parms._response_column), score.vec("predict")); _job.update(1); _model.update(_job); + LOG.info((n + 1) + ". estimator was built in " + timer.toString()); + LOG.info("*********************************************************************"); } if (_trainWithWeights != _parms.train()) { DKV.remove(_trainWithWeights._key);