readRDS.lgb.Booster: readRDS for 'lgb.Booster' models

In lightgbm: Light Gradient Boosting Machine

readRDS for lgb.Booster models

Description

Attempts to load a model stored in a .rds file, using readRDS

Usage

readRDS.lgb.Booster(file, refhook = NULL)

Arguments

file	a connection or the name of the file where the R object is saved to or read from.
refhook	a hook function for handling reference objects.

Value

lgb.Booster

Examples

library(lightgbm)
data(agaricus.train, package = "lightgbm")
train <- agaricus.train
dtrain <- lgb.Dataset(train$data, label = train$label)
data(agaricus.test, package = "lightgbm")
test <- agaricus.test
dtest <- lgb.Dataset.create.valid(dtrain, test$data, label = test$label)
params <- list(
  objective = "regression"
  , metric = "l2"
  , min_data = 1L
  , learning_rate = 1.0
)
valids <- list(test = dtest)
model <- lgb.train(
  params = params
  , data = dtrain
  , nrounds = 10L
  , valids = valids
  , early_stopping_rounds = 5L
)
model_file <- tempfile(fileext = ".rds")
saveRDS.lgb.Booster(model, model_file)
new_model <- readRDS.lgb.Booster(model_file)

lightgbm documentation built on Jan. 17, 2023, 1:13 a.m.