h2o_rulefit_train: Wrapper for training a h2o.rulefit model as part of a parsnip...

View source: R/rule_fit.R

h2o_rulefit_trainR Documentation

Wrapper for training a h2o.rulefit model as part of a parsnip 'rule_fit' h2o engine

Description

Wrapper for training a h2o.rulefit model as part of a parsnip 'rule_fit' h2o engine

Usage

h2o_rulefit_train(
  formula,
  data,
  rule_generation_ntrees = 50,
  max_rule_length = 3,
  lambda = 0,
  ...
)

Arguments

formula

formula

data

data.frame of training data

rule_generation_ntrees

integer, the number of trees to build (default = 50)

max_rule_length

integer, the maximum tree depth (default = 3).

lambda

Specify the regularization strength for LASSO regressor.

...

other arguments that are passed to the h2o model

Value

evaluated h2o model call


stevenpawley/h2oparsnip documentation built on June 20, 2022, 12:48 p.m.