Skip to contents

If .data is empty an InducerXGBoost object will be created. If .data is a Dataset object a xgboost model will be fitted

Usage

InducerXGBoost(
  .data = NULL,
  nrounds = 1,
  eta = 0.3,
  gamma = 0,
  max_depth = 6,
  min_child_weight = 1,
  subsample = 1,
  colsample_bytree = 1,
  lambda = 1,
  alpha = 0,
  num_parallel_tree = 1
)

Arguments

.data

Data object of class Dataset.

nrounds

number of rounds

eta

eta value

gamma

gamma

max_depth

max depth paramater

min_child_weight

min child weight paramater

subsample

subsample paramater

colsample_bytree

colsample paramater

lambda

lambda paramater

alpha

alpha paramater

num_parallel_tree

number of parallel tree paramater

Value

A InducerXGBoost object

Examples

inducer <- InducerXGBoost()
inducer
#> Inducer: XGBoost
#> Configuration:  nrounds = 1, eta = 0.3, gamma = 0, max_depth = 6, min_child_weight = 1, subsample = 1, colsample_bytree = 1, lambda = 1, alpha = 0, num_parallel_tree = 1
cars.data <- Dataset(data = cars, target = "dist")
fittedInd <- InducerXGBoost(.data = cars.data)
#> [1]	train-rmse:37.257189 
fittedInd
#> Regression Model: "InducerXGBoost" fitted on "cars" dataset.