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.

Apache License 2.0

6.86k stars 1.99k forks source link

Let's add a include_default_model argument (defaulting to TRUE) to h2o.grid. I always want to compare my grid search models to the default model and currently you have to write 2x the amount of code to do this:

{code}

Add a default DNN to the grid

dl_grid <- h2o.grid(algorithm = "deeplearning", grid_id = randomgrid_name, x = x, y = y, training_frame = train, nfolds = nfolds, keep_cross_validation_predictions = TRUE, seed = seed, hyper_params = list(activation = "Rectifier"), #dummy search_criteria = list(strategy = "RandomDiscrete", max_models = 1))

Train Random Grid

dl_grid <- h2o.grid(algorithm = "deeplearning", grid_id = randomgrid_name, x = x, y = y, training_frame = train, nfolds = nfolds, keep_cross_validation_predictions = TRUE, seed = seed, hyper_params = hyper_params, search_criteria = search_criteria) {code}

h2oai / h2o-3

Add an include_default_model argument to grid search #11527

Add a default DNN to the grid

Train Random Grid