setup_LightGBM: Setup LightGBM Hyperparameters
In egenn/rtemis: Machine Learning and Visualization

setup_LightGBM

R Documentation

Setup LightGBM Hyperparameters

Description

Setup hyperparameters for LightGBM training.

Usage

setup_LightGBM(
  max_nrounds = 500L,
  force_nrounds = NULL,
  early_stopping_rounds = 10L,
  num_leaves = 8L,
  max_depth = -1L,
  learning_rate = 0.01,
  feature_fraction = 1,
  subsample = 1,
  subsample_freq = 1L,
  lambda_l1 = 0,
  lambda_l2 = 0,
  max_cat_threshold = 32L,
  min_data_per_group = 32L,
  linear_tree = FALSE,
  ifw = FALSE,
  objective = NULL,
  device_type = "cpu",
  tree_learner = "serial",
  force_col_wise = TRUE,
  num_threads = 0L
)

Arguments

`max_nrounds`	Positive integer: Maximum number of boosting rounds.
`force_nrounds`	Positive integer: Use this many boosting rounds. Disable search for nrounds.
`early_stopping_rounds`	Positive integer: Number of rounds without improvement to stop training.
`num_leaves`	(Tunable) Positive integer: Maximum number of leaves in one tree.
`max_depth`	(Tunable) Integer: Maximum depth of trees.
`learning_rate`	(Tunable) Numeric: Learning rate.
`feature_fraction`	(Tunable) Numeric: Fraction of features to use.
`subsample`	(Tunable) Numeric: Fraction of data to use.
`subsample_freq`	(Tunable) Positive integer: Frequency of subsample.
`lambda_l1`	(Tunable) Numeric: L1 regularization.
`lambda_l2`	(Tunable) Numeric: L2 regularization.
`max_cat_threshold`	(Tunable) Positive integer: Maximum number of categories for categorical features.
`min_data_per_group`	(Tunable) Positive integer: Minimum number of observations per categorical group.
`linear_tree`	Logical: If TRUE, use linear trees.
`ifw`	Logical: If TRUE, use Inverse Frequency Weighting in classification.
`objective`	Character: Objective function.
`device_type`	Character: "cpu" or "gpu".
`tree_learner`	Character: "serial", "feature", "data", or "voting".
`force_col_wise`	Logical: Use only with CPU - If TRUE, force col-wise histogram building
`num_threads`	Integer: Number of threads to use. 0 means default number of threads in OpenMP.