Skip to content

Commit 00cabd9

Browse files
committed
Fix predict_numeric error message on classification models
There is no `predict_prob` function.
1 parent 1f96393 commit 00cabd9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

R/predict_numeric.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
predict_numeric.model_fit <- function (object, new_data, ...) {
99
if (object$spec$mode != "regression")
1010
stop("`predict_numeric` is for predicting numeric outcomes. ",
11-
"Use `predict_class` or `predict_prob` for ",
11+
"Use `predict_class` or `predict_classprob` for ",
1212
"classification models.", call. = FALSE)
1313

1414
if (!any(names(object$spec$method) == "numeric"))

0 commit comments

Comments
 (0)