"docs/_static/images/dask-initial-setup.svg" did not exist on "296397df7bf584770a0297d84ce6a8f4ef25c317"
lgb.load.Rd 1.2 KB
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
% 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}
}
\value{
15
lgb.Booster
Guolin Ke's avatar
Guolin Ke committed
16
17
}
\description{
18
19
Load LightGBM takes in either a file path or model string.
              If both are provided, Load will default to loading from file
Guolin Ke's avatar
Guolin Ke committed
20
21
22
23
24
25
26
27
28
29
30
}
\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")
valids <- list(test = dtest)
31
32
33
model <- lgb.train(
  params = params
  , data = dtrain
34
  , nrounds = 10L
35
  , valids = valids
36
37
38
  , min_data = 1L
  , learning_rate = 1.0
  , early_stopping_rounds = 5L
39
)
Guolin Ke's avatar
Guolin Ke committed
40
lgb.save(model, "model.txt")
41
42
43
load_booster <- lgb.load(filename = "model.txt")
model_string <- model$save_model_to_string(NULL) # saves best iteration
load_booster_from_str <- lgb.load(model_str = model_string)
Guolin Ke's avatar
Guolin Ke committed
44
45

}