RemixAutoML (version 0.4.2)

XGBoostParameterGrids: XGBoostParameterGrids

Description

XGBoostParameterGrids

Usage

XGBoostParameterGrids(
  TaskType = "CPU",
  Shuffles = 1L,
  NTrees = seq(500L, 5000L, 500L),
  Depth = seq(4L, 16L, 2L),
  LearningRate = seq(0.05, 0.4, 0.05),
  MinChildWeight = seq(1, 10, 1),
  SubSample = seq(0.55, 1, 0.05),
  ColSampleByTree = seq(0.55, 1, 0.05)
)

Arguments

TaskType

"GPU" or "CPU"

Shuffles

The number of shuffles you want to apply to each grid

NTrees

seq(500L, 5000L, 500L)

Depth

seq(4L, 16L, 2L)

LearningRate

seq(0.05,0.40,0.05)

MinChildWeight

seq(1.0, 10.0, 1.0)

SubSample

seq(0.55, 1.0, 0.05)

ColSampleByTree

seq(0.55, 1.0, 0.05)

Value

A list containing data.table's with the parameters shuffled and ready to test in the bandit framework

See Also

Other Supervised Learning: AutoH2OScoring(), CatBoostClassifierParams(), CatBoostMultiClassParams(), CatBoostParameterGrids(), CatBoostRegressionParams(), XGBoostClassifierParams(), XGBoostMultiClassParams(), XGBoostRegressionMetrics(), XGBoostRegressionParams()