replace nround with nrounds to match actual parameter (#3592)
This commit is contained in:
committed by
Philip Hyunsu Cho
parent
73bd590a1d
commit
725f4c36f2
@@ -42,7 +42,7 @@ R
|
||||
train <- agaricus.train
|
||||
test <- agaricus.test
|
||||
# fit model
|
||||
bst <- xgboost(data = train$data, label = train$label, max.depth = 2, eta = 1, nround = 2,
|
||||
bst <- xgboost(data = train$data, label = train$label, max.depth = 2, eta = 1, nrounds = 2,
|
||||
nthread = 2, objective = "binary:logistic")
|
||||
# predict
|
||||
pred <- predict(bst, test$data)
|
||||
|
||||
Reference in New Issue
Block a user