| xgb.save | R Documentation |
Save XGBoost model to a file in binary or JSON format.
xgb.save(model, fname)
model |
Model object of |
fname |
Name of the file to write. Its extension determines the serialization format:
|
This methods allows to save a model in an XGBoost-internal binary or text format which is universal
among the various xgboost interfaces. In R, the saved model file could be read later
using either the xgb.load() function or the xgb_model parameter of xgb.train().
Note: a model can also be saved as an R object (e.g., by using readRDS()
or save()). However, it would then only be compatible with R, and
corresponding R methods would need to be used to load it. Moreover, persisting the model with
readRDS() or save() might cause compatibility problems in
future versions of XGBoost. Consult a-compatibility-note-for-saveRDS-save to learn
how to persist models in a future-proof way, i.e., to make the model accessible in future
releases of XGBoost.
xgb.load()
data(agaricus.train, package = "xgboost")
data(agaricus.test, package = "xgboost")
## Keep the number of threads to 1 for examples
nthread <- 1
data.table::setDTthreads(nthread)
train <- agaricus.train
test <- agaricus.test
bst <- xgb.train(
data = xgb.DMatrix(train$data, label = train$label, nthread = 1),
nrounds = 2,
params = xgb.params(
max_depth = 2,
nthread = nthread,
objective = "binary:logistic"
)
)
fname <- file.path(tempdir(), "xgb.ubj")
xgb.save(bst, fname)
bst <- xgb.load(fname)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.