Learn R Programming

fastml (version 0.2.0)

define_lightgbm_spec: Define LightGBM Model Specification

Description

Define LightGBM Model Specification

Usage

define_lightgbm_spec(task, train_data, label, tune = FALSE)

Value

List containing the model specification (`model_spec`).

Arguments

task

Character string specifying the task type: "classification" or "regression".

train_data

Data frame containing the training data.

label

Character string specifying the name of the target variable.

tune

Logical indicating whether to use tuning parameters.