Turn off early stopping in DRF and XRT in AutoML

h2oai / h2o-3

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

http://h2o.ai

Apache License 2.0

6.85k stars 1.99k forks source link

Turn off early stopping in DRF and XRT in AutoML #12736

Open exalate-issue-sync[bot] opened 1 year ago

exalate-issue-sync[bot] commented 1 year ago

Consider turning off early stopping in the default DRF and XRT models in AutoML since we only have 50 trees. First, evaluate on some datasets and see how this affects performance. Let's benchmark to see if this is a good idea or not.

One issue here is that the default score iteration is not an even number -- another option is to change score_tree_interval to 2 or 5.

hasithjp commented 1 year ago

JIRA Issue Migration Info

Jira Issue: PUBDEV-5884 Assignee: Sebastien Poirier Reporter: Erin LeDell State: Open Fix Version: N/A Attachments: N/A Development PRs: N/A