xgb.save {xgboost} | R Documentation |
Save xgboost model from xgboost or xgb.train
xgb.save(model, fname)
model |
the model object. |
fname |
the name of the file to write. |
data(agaricus.train, package='xgboost') data(agaricus.test, package='xgboost') train <- agaricus.train test <- agaricus.test bst <- xgboost(data = train$data, label = train$label, max_depth = 2, eta = 1, nthread = 2, nrounds = 2,objective = "binary:logistic") xgb.save(bst, 'xgb.model') bst <- xgb.load('xgb.model') pred <- predict(bst, test$data)