Skip to content

Commit f3bee97

Browse files
committed
Fix two warnings in boosted tree C5.0 tests
1 parent b23ae2b commit f3bee97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/testthat/test_boost_tree_C50.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,9 +120,9 @@ test_that('submodel prediction', {
120120
data = wa_churn[-(1:4), c("churn", vars)],
121121
engine = "C5.0")
122122

123-
pred_class <- predict(class_fit$fit, wa_churn[1:4, vars], trials = 5, type = "prob")
123+
pred_class <- predict(class_fit$fit, wa_churn[1:4, vars], trials = 4, type = "prob")
124124

125-
mp_res <- multi_predict(class_fit, new_data = wa_churn[1:4, vars], trees = 5, type = "prob")
125+
mp_res <- multi_predict(class_fit, new_data = wa_churn[1:4, vars], trees = 4, type = "prob")
126126
mp_res <- do.call("rbind", mp_res$.pred)
127127
expect_equal(mp_res[[".pred_No"]], unname(pred_class[, "No"]))
128128
})

0 commit comments

Comments
 (0)