Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[R] Replace xgboost() with xgb.train() in most tests and examples (#9941)

Browse Source
This commit is contained in:
david-cortes
2024-01-02 14:20:01 +01:00
committed by GitHub
parent 32cbab1cc0
commit 9e33a10202
27 changed files with 156 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
R-package/man/a-compatibility-note-for-saveRDS-save.Rd
View File

@@ -33,8 +33,9 @@ For more details and explanation about model persistence and archival, consult t
}
\examples{
data(agaricus.train, package='xgboost')
bst <- xgboost(data = agaricus.train$data, label = agaricus.train$label, max_depth = 2,
eta = 1, nthread = 2, nrounds = 2, objective = "binary:logistic")
bst <- xgb.train(data = xgb.DMatrix(agaricus.train$data, label = agaricus.train$label),
max_depth = 2, eta = 1, nthread = 2, nrounds = 2,
objective = "binary:logistic")
# Save as a stand-alone file; load it with xgb.load()
xgb.save(bst, 'xgb.model')

15
R-package/man/predict.xgb.Booster.Rd
View File

@@ -136,9 +136,8 @@ data.table::setDTthreads(nthread)
train <- agaricus.train
test <- agaricus.test
bst <- xgboost(
data = train$data,
label = train$label,
bst <- xgb.train(
data = xgb.DMatrix(train$data, label = train$label),
max_depth = 2,
eta = 0.5,
nthread = nthread,
@@ -180,9 +179,8 @@ num_class <- 3
set.seed(11)
bst <- xgboost(
data = as.matrix(iris[, -5]),
label = lb,
bst <- xgb.train(
data = xgb.DMatrix(as.matrix(iris[, -5]), label = lb),
max_depth = 4,
eta = 0.5,
nthread = 2,
@@ -205,9 +203,8 @@ sum(pred_labels != lb) / length(lb)
# compare with predictions from softmax:
set.seed(11)
bst <- xgboost(
data = as.matrix(iris[, -5]),
label = lb,
bst <- xgb.train(
data = xgb.DMatrix(as.matrix(iris[, -5]), label = lb),
max_depth = 4,
eta = 0.5,
nthread = 2,

6
R-package/man/xgb.load.Rd
View File

@@ -34,8 +34,10 @@ data.table::setDTthreads(nthread)
train <- agaricus.train
test <- agaricus.test
bst <- xgboost(
data = train$data, label = train$label, max_depth = 2, eta = 1,
bst <- xgb.train(
data = xgb.DMatrix(train$data, label = train$label),
max_depth = 2,
eta = 1,
nthread = nthread,
nrounds = 2,
objective = "binary:logistic"

6
R-package/man/xgb.save.Rd
View File

@@ -38,8 +38,10 @@ data.table::setDTthreads(nthread)
train <- agaricus.train
test <- agaricus.test
bst <- xgboost(
data = train$data, label = train$label, max_depth = 2, eta = 1,
bst <- xgb.train(
data = xgb.DMatrix(train$data, label = train$label),
max_depth = 2,
eta = 1,
nthread = nthread,
nrounds = 2,
objective = "binary:logistic"

4
R-package/man/xgb.save.raw.Rd
View File

@@ -32,8 +32,8 @@ data.table::setDTthreads(nthread)
train <- agaricus.train
test <- agaricus.test
bst <- xgboost(data = train$data, label = train$label, max_depth = 2,
eta = 1, nthread = nthread, nrounds = 2,objective = "binary:logistic")
bst <- xgb.train(data = xgb.DMatrix(train$data, label = train$label), max_depth = 2,
eta = 1, nthread = nthread, nrounds = 2,objective = "binary:logistic")
raw <- xgb.save.raw(bst)
bst <- xgb.load.raw(raw)

4
R-package/man/xgb.serialize.Rd
View File

@@ -21,8 +21,8 @@ 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")
bst <- xgb.train(data = xgb.DMatrix(train$data, label = train$label), max_depth = 2,
eta = 1, nthread = 2, nrounds = 2,objective = "binary:logistic")
raw <- xgb.serialize(bst)
bst <- xgb.unserialize(raw)