-
Notifications
You must be signed in to change notification settings - Fork 88
/
predict_raw.R
42 lines (33 loc) · 1.04 KB
/
predict_raw.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#' @rdname predict.model_fit
#' @method predict_raw model_fit
#' @export predict_raw.model_fit
#' @export
predict_raw.model_fit <- function(object, new_data, opts = list(), ...) {
protected_args <- names(object$spec$method$pred$raw$args)
dup_args <- names(opts) %in% protected_args
if (any(dup_args)) {
opts <- opts[[!dup_args]]
}
if (length(opts) > 0) {
object$spec$method$pred$raw$args <-
c(object$spec$method$pred$raw$args, opts)
}
check_spec_pred_type(object, "raw")
if (inherits(object$fit, "try-error")) {
cli::cli_warn("Model fit failed; cannot make predictions.")
return(NULL)
}
new_data <- prepare_data(object, new_data)
# preprocess data
if (!is.null(object$spec$method$pred$raw$pre))
new_data <- object$spec$method$pred$raw$pre(new_data, object)
# create prediction call
pred_call <- make_pred_call(object$spec$method$pred$raw)
res <- eval_tidy(pred_call)
res
}
#' @export
#' @keywords internal
#' @rdname predict.model_fit
predict_raw <- function(object, ...)
UseMethod("predict_raw")