LightGBM Auto

Lightgbm auto.

Value

Object of class R6::R6Class and AutoLightGBM.

Super class

Auto -> AutoLightGBM

Methods

Public methods

• AutoLightGBM$new()

• AutoLightGBM$graph()

• AutoLightGBM$finalize_model()

• AutoLightGBM$estimate_memory()

• AutoLightGBM$internal_measure()

• AutoLightGBM$clone()

Inherited methods

• Auto$check()
• Auto$design_default()
• Auto$design_set()
• Auto$early_stopping_rounds()
• Auto$search_space()

---

AutoLightGBM$new()

Creates a new instance of this R6 class.

Usage

AutoLightGBM$new(id = "lightgbm")

Arguments

id

(character(1))
Identifier for the new instance.

---

AutoLightGBM$graph()

Create the graph for the auto.

Usage

AutoLightGBM$graph(task, measure, n_threads, timeout, devices)

Arguments

task

(mlr3::Task).

measure

(mlr3::Measure).

n_threads

(integer(1)).

timeout

(integer(1)).

devices

(character())
Devices to use. Allowed values are "cpu" and "cuda". Default is "cpu".

---

AutoLightGBM$finalize_model()

Prepare the graph learner for the final model fit.

Usage

AutoLightGBM$finalize_model(graph_learner)

Arguments

graph_learner

(mlr3pipelines::GraphLearner).

---

AutoLightGBM$estimate_memory()

Estimate the memory for the auto.

Usage

AutoLightGBM$estimate_memory(task)

Arguments

task

(mlr3::Task).

---

AutoLightGBM$internal_measure()

Get the internal measure for the auto.

Usage

AutoLightGBM$internal_measure(measure, task)

Arguments

measure

(mlr3::Measure).

task

(mlr3::Task).

---

AutoLightGBM$clone()

The objects of this class are cloneable with this method.

Usage

AutoLightGBM$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.

Examples

auto("lightgbm")
#> <AutoLightGBM>
#>   Inherits from: <Auto>
#>   Public:
#>     check: function (task, memory_limit = Inf, large_data_set = FALSE, devices) 
#>     clone: function (deep = FALSE) 
#>     design_default: function (task) 
#>     design_set: function (task, measure, size) 
#>     devices: cpu cuda
#>     early_stopping_rounds: function (task, budget = Inf) 
#>     estimate_memory: function (task) 
#>     finalize_model: function (graph_learner) 
#>     graph: function (task, measure, n_threads, timeout, devices) 
#>     id: lightgbm
#>     initialize: function (id = "lightgbm") 
#>     internal_measure: function (measure, task) 
#>     packages: mlr3 mlr3extralearners lightgbm
#>     properties: internal_tuning large_data_sets
#>     search_space: function (task) 
#>     task_types: classif regr
#>   Private:
#>     .default_values: list
#>     .search_space: ParamSet, R6