fix java api

h2oai · Sep 8, 2023 · 6f024ec · 6f024ec
1 parent f8ffdb9
commit 6f024ec
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/h2o-algos/src/main/java/hex/adaboost/AdaBoost.java b/h2o-algos/src/main/java/hex/adaboost/AdaBoost.java
@@ -7,6 +7,7 @@
 import hex.glm.GLMModel;
 import hex.tree.drf.DRF;
 import hex.tree.drf.DRFModel;
+import hex.tree.dt.DTModel;
 import org.apache.log4j.Logger;
 import water.*;
 import water.exceptions.H2OModelBuilderIllegalArgumentException;
@@ -33,6 +34,20 @@ public AdaBoost(AdaBoostModel.AdaBoostParameters parms) {
         init(false);
     }
 
+    public AdaBoost(boolean startup_once) {
+        super(new AdaBoostModel.AdaBoostParameters(), startup_once);
+    }
+
+    @Override
+    public boolean havePojo() {
+        return false;
+    }
+
+    @Override
+    public boolean haveMojo() {
+        return false;
+    }
+
     @Override
     public void init(boolean expensive) {
         super.init(expensive);

diff --git a/h2o-algos/src/main/java/hex/adaboost/AdaBoostModel.java b/h2o-algos/src/main/java/hex/adaboost/AdaBoostModel.java
@@ -92,7 +92,7 @@ public static class AdaBoostParameters extends Model.Parameters {
         /**
          * TODO valenad1
          */
-        public long _n_estimators;
+        public int _n_estimators;
 
         /**
          * TODO valenad1