Setup hyperparameters for LightCART training.
Usage
setup_LightCART(
num_leaves = 32L,
max_depth = -1L,
lambda_l1 = 0,
lambda_l2 = 0,
max_cat_threshold = 32L,
min_data_per_group = 100L,
linear_tree = FALSE,
objective = NULL,
ifw = FALSE
)Arguments
- num_leaves
(Tunable) Positive integer: Maximum number of leaves in one tree.
- max_depth
(Tunable) Integer: Maximum depth of trees.
- 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 data per categorical group.
- linear_tree
(Tunable) Logical: If TRUE, use linear trees.
- objective
Character: Objective function.
- ifw
Logical: If TRUE, use Inverse Frequency Weighting in classification.
Details
Get more information from lightgbm::lgb.train.