makeMrAshLearner: Create a mr.ash learner object

View source: R/mr_ash_learner.R

makeMrAshLearnerR Documentation

Create a mr.ash learner object

Description

Creates a mr.ash learner object to be used in veb_boost

Usage

makeMrAshLearner(
  X,
  X_test = NULL,
  growMode = c("NA", "+*", "+", "*"),
  changeToConstant = FALSE
)

Arguments

X

is a matrix to be used as the predictors in training. No scaling is performed by the function, so the user must do their own scaling (scaling so that each column has a standard deviation of 1 is recommended)

X_test

is a matrix to be used as the predictors in testing. No scaling is performed by the function, so the user must do their own scaling (scaling so that each column has a standard deviation of 1 is recommended)

growMode

is a string for if the learner should be grown (or not) If "+*", we grow mu_0 -> (mu_0 * mu_2) + mu_1 If "+", we grow mu_0 -> (mu_0 + mu_1) If "*", we grow mu_0 -> (mu_0 * mu_1) (NOTE: Not recommended if we start with k = 1) If "NA", we do not grow this learner

changeToConstant

is a logical for if constant fits should be changed to be constant


stephenslab/VEB.Boost documentation built on July 2, 2023, 1 p.m.