wehub-resource-sync
53 lines
1.4 KiB
R
53 lines
1.4 KiB
R
% Generated by roxygen2: do not edit by hand
|
|
% Please edit documentation in R/lgb.Booster.R
|
|
\name{lgb.load}
|
|
\alias{lgb.load}
|
|
\title{Load LightGBM model}
|
|
\usage{
|
|
lgb.load(filename = NULL, model_str = NULL)
|
|
}
|
|
\arguments{
|
|
\item{filename}{path of model file}
|
|
|
|
\item{model_str}{a str containing the model (as a \code{character} or \code{raw} vector)}
|
|
}
|
|
\value{
|
|
lgb.Booster
|
|
}
|
|
\description{
|
|
Load LightGBM takes in either a file path or model string.
|
|
If both are provided, Load will default to loading from file
|
|
}
|
|
\examples{
|
|
\donttest{
|
|
\dontshow{setLGBMthreads(2L)}
|
|
\dontshow{data.table::setDTthreads(1L)}
|
|
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
|
|
, num_threads = 2L
|
|
)
|
|
valids <- list(test = dtest)
|
|
model <- lgb.train(
|
|
params = params
|
|
, data = dtrain
|
|
, nrounds = 5L
|
|
, valids = valids
|
|
, early_stopping_rounds = 3L
|
|
)
|
|
model_file <- tempfile(fileext = ".txt")
|
|
lgb.save(model, model_file)
|
|
load_booster <- lgb.load(filename = model_file)
|
|
model_string <- model$save_model_to_string(NULL) # saves best iteration
|
|
load_booster_from_str <- lgb.load(model_str = model_string)
|
|
}
|
|
}
|