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

[R] resolve assignment_linter warnings (#8599)

Browse Source
This commit is contained in:
James Lamb
2022-12-16 11:22:41 -06:00
committed by GitHub
parent f6effa1734
commit 53e6e32718
8 changed files with 43 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
demo/multiclass_classification/train.R
View File

@@ -26,7 +26,7 @@ test_x <- test[, 1:34]
test_y <- test[, V35]
xg_train <- xgb.DMatrix(data = as.matrix(train_x), label = train_y)
xg_test = xgb.DMatrix(as.matrix(test_x), label = test_y)
xg_test <- xgb.DMatrix(as.matrix(test_x), label = test_y)
params <- list(
objective = 'multi:softmax',
@@ -36,7 +36,7 @@ params <- list(
eta = 0.1
)
watchlist = list(train = xg_train, test = xg_test)
watchlist <- list(train = xg_train, test = xg_test)
bst <- xgb.train(
params = params,
@@ -60,5 +60,5 @@ pred_mat <- matrix(pred_prob, ncol = 6, byrow = TRUE)
# rowSums(pred_mat)
pred_label <- apply(pred_mat, 1, which.max) - 1L
error_rate = sum(pred_label != test_y) / length(test_y)
error_rate <- sum(pred_label != test_y) / length(test_y)
print(paste("Test error using softprob =", error_rate))