dl_autogrid: dl_autogrid

Description Usage Arguments Value

Description

dl_autogrid is a wrapper employing built-in settings to run grid search hyper parameter optimizations on deep learning (dl_ algorithms)

Usage

1
2
3
4
5
6
dl_autogrid(train, valid, y, x, eval_metric = "AUTO", wd = getwd(),
  folds = NULL, deeplearning_runtime_secs = 10,
  deeplearning_stopping_rounds = 10,
  deeplearning_stopping_tolerance = 1e-05,
  deeplearning_adaptive_rate = TRUE, grid_strategy = "RandomDiscrete",
  grid_id = "dl")

Arguments

train

H2O frame object containing labeled data for model training. No Default.

valid

H2O frame object containing labeled data for model validation. No Default.

y

Character object of length 1 identifying the column name of the target variable. No Default.

x

Character object of length 1 or more identifying the column name(s) of the input variables. No Default.

eval_metric

Character object defining evaluation metric for training. Defualt is "AUTO" and uses built-in H2O automatic choice for target data type.

wd

Character object defining file path where dl_models folder will be created and deep learning models saved. Defaults to current working directory.

folds

Character object defining number of folds for xval. Default is NULL and currently is not implemented.

deeplearning_runtime_secs

Numeric object defining total number of seconds the hyper parameter grid search will run.

deeplearning_stopping_rounds

Numeric object defining maximum number of training rounds an individual deep learning model not improving will continue to run. Default is 10.

deeplearning_stopping_tolerance

Numeric object which sets the mimmum loss funciton improvement for a training iteration to be considered an improvement. Defulat is 1E-5.

deeplearning_adaptive_rate

Boolean, if TRUE ADELTA is used to control learning rate if FALSE than normal rate controls can be used.

grid_strategy

Character object default and only current supported option is "randomDiscrete"

grid_id

Character. Set grid name in h2o platform. Defualts to "dl" only used in ensemble function.

Value

List object containing H2O model objects. Additionally saves h2o models as re-loadable text files in wd/dl_models folder.


andrewsommerlot/startml documentation built on May 5, 2019, 6:58 p.m.