Use non default parameters in smoke test

h2oai · Sep 22, 2023 · 72b2298 · 72b2298
1 parent fc7cea4
commit 72b2298
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/h2o-py/tests/testdir_algos/adaboost/pyunit_adaboost_smoke.py b/h2o-py/tests/testdir_algos/adaboost/pyunit_adaboost_smoke.py
@@ -11,7 +11,7 @@ def adaboost():
     train = h2o.import_file(pyunit_utils.locate("smalldata/prostate/prostate.csv"))
     train["CAPSULE"] = train["CAPSULE"].asfactor()
 
-    adaboost_model = H2OAdaBoostEstimator(nlearners=50, seed=0xBEEF, weak_learner="DRF", learn_rate=0.5)
+    adaboost_model = H2OAdaBoostEstimator(nlearners=55, seed=0xBEEF, weak_learner="GLM", learn_rate=0.6)
     adaboost_model.train(training_frame=train, y="CAPSULE")
     predict = adaboost_model.predict(train)
 
@@ -20,9 +20,9 @@ def adaboost():
     print("")
     print(predict)
 
-    assert 50 == adaboost_model._model_json["output"]["model_summary"]["number_of_weak_learners"][0], "Python API is not working!"
-    assert "DRF" == adaboost_model._model_json["output"]["model_summary"]["weak_learner"][0], "Python API is not working!"
-    assert 0.5 == adaboost_model._model_json["output"]["model_summary"]["learn_rate"][0], "Python API is not working!"
+    assert 55 == adaboost_model._model_json["output"]["model_summary"]["number_of_weak_learners"][0], "Python API is not working!"
+    assert "GLM" == adaboost_model._model_json["output"]["model_summary"]["weak_learner"][0], "Python API is not working!"
+    assert 0.6 == adaboost_model._model_json["output"]["model_summary"]["learn_rate"][0], "Python API is not working!"
 
 
 if __name__ == "__main__":