optimize_gbm_parameters: Optimizing GBM parameters.

Description Usage Arguments Value

View source: R/h2o_optimize.R

Description

Optimazation is done in two steps: (1) maximum depth is optimized, (2) the rest of the parameters are optimized. This procedure makes the optimization problem more manageable.

Usage

1
2
3
4
5
6
7
8
9
optimize_gbm_parameters(
  h2o_data,
  set,
  algorithm,
  metric,
  predictors,
  targets,
  runid
)

Arguments

h2o_data

a list of H2OFrames

set

config object

algorithm

either 'gbm' or 'xgboost'

metric

evaluation metric

predictors

vector of explanatory variables

targets

label variable

runid

run ID

Value

best model


Bilot/AI-jack-opensource-R documentation built on July 28, 2020, 1:15 a.m.