readRDS.lgb.Booster.R 1.68 KB
Newer Older
1
#' @name readRDS.lgb.Booster
2
3
4
5
6
#' @title readRDS for \code{lgb.Booster} models (DEPRECATED)
#' @description Calls \code{readRDS} in what is expected to be a serialized \code{lgb.Booster} object,
#'              and then restores its handle through \code{lgb.restore_handle}.
#'
#'              \bold{This function throws a warning and will be removed in future versions.}
7
8
#' @param file a connection or the name of the file where the R object is saved to or read from.
#' @param refhook a hook function for handling reference objects.
9
#'
10
#' @return \code{lgb.Booster}
11
#'
12
#' @examples
13
#' \donttest{
14
#' library(lightgbm)
15
16
#' \dontshow{setLGBMthreads(2L)}
#' \dontshow{data.table::setDTthreads(1L)}
17
18
19
20
21
22
#' 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)
23
24
25
26
27
#' params <- list(
#'   objective = "regression"
#'   , metric = "l2"
#'   , min_data = 1L
#'   , learning_rate = 1.0
28
#'   , num_threads = 2L
29
#' )
30
#' valids <- list(test = dtest)
31
32
33
#' model <- lgb.train(
#'   params = params
#'   , data = dtrain
34
#'   , nrounds = 10L
35
#'   , valids = valids
36
#'   , early_stopping_rounds = 5L
37
#' )
38
39
40
#' model_file <- tempfile(fileext = ".rds")
#' saveRDS.lgb.Booster(model, model_file)
#' new_model <- readRDS.lgb.Booster(model_file)
41
#' }
42
#' @export
43
readRDS.lgb.Booster <- function(file, refhook = NULL) {
44

45
  warning("'readRDS.lgb.Booster' is deprecated and will be removed in a future release. Use readRDS() instead.")
46

47
48
49
  object <- readRDS(file = file, refhook = refhook)
  lgb.restore_handle(object)
  return(object)
50
}