Skip to content

Commit 6244c1d

Browse files
committed
fix test: "Yes" is the first level, thus the event level
1 parent 1e945f4 commit 6244c1d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/testthat/test_boost_tree_xgboost.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ test_that('submodel prediction', {
210210

211211
mp_res <- multi_predict(class_fit, new_data = wa_churn[1:4, vars], trees = 5, type = "prob")
212212
mp_res <- do.call("rbind", mp_res$.pred)
213-
expect_equal(mp_res[[".pred_No"]], pred_class)
213+
expect_equal(mp_res[[".pred_Yes"]], pred_class)
214214

215215
expect_error(
216216
multi_predict(class_fit, newdata = wa_churn[1:4, vars], trees = 5, type = "prob"),

0 commit comments

Comments
 (0)