ht/fix

h2oai · Jan 31, 2024 · 6a76d36 · 6a76d36
1 parent df77e31
commit 6a76d36
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/h2o-bindings/bin/custom/python/gen_modelselection.py b/h2o-bindings/bin/custom/python/gen_modelselection.py
@@ -276,7 +276,7 @@ def get_best_model_predictors(self):
 >>> maxrModel.train(x=predictors, y=response, training_frame=prostate)
 >>> result = maxrModel.result()
 >>> # get the GLM model with the best performance for a fixed predictor size:
->>> one_model = h2o.get_model(result["model_id"][ind,0])
+>>> one_model = h2o.get_model(result["model_id"][1, 0])
 >>> predict = one_model.predict(prostate)
 >>> # print a version of the predict frame:
 >>> print(predict)

diff --git a/h2o-py/h2o/estimators/model_selection.py b/h2o-py/h2o/estimators/model_selection.py
@@ -1238,7 +1238,7 @@ def build_glm_model(self):
         >>> maxrModel.train(x=predictors, y=response, training_frame=prostate)
         >>> result = maxrModel.result()
         >>> # get the GLM model with the best performance for a fixed predictor size:
-        >>> one_model = h2o.get_model(result["model_id"][ind,0])
+        >>> one_model = h2o.get_model(result["model_id"][1, 0])
         >>> predict = one_model.predict(prostate)
         >>> # print a version of the predict frame:
         >>> print(predict)