predict_quantile.R

#' @keywords internal
#' @rdname other_predict
#' @param quantile A vector of numbers between 0 and 1 for the quantile being
#'  predicted.
#' @inheritParams predict.model_fit
#' @method predict_quantile model_fit
#' @export predict_quantile.model_fit
#' @export
predict_quantile.model_fit <- function(object,
                                       new_data,
                                       quantile = (1:9)/10,
                                       interval = "none",
                                       level = 0.95,
                                       ...) {

  check_spec_pred_type(object, "quantile")

  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$quantile$pre)) {
    new_data <- object$spec$method$pred$quantile$pre(new_data, object)
  }

  # Pass some extra arguments to be used in post-processor
  object$spec$method$pred$quantile$args$p <- quantile
  pred_call <- make_pred_call(object$spec$method$pred$quantile)

  res <- eval_tidy(pred_call)

  # post-process the predictions
  if(!is.null(object$spec$method$pred$quantile$post)) {
    res <- object$spec$method$pred$quantile$post(res, object)
  }

  res
}

# @export
# @keywords internal
# @rdname other_predict
# @inheritParams predict.model_fit
predict_quantile <- function (object, ...) {
  UseMethod("predict_quantile")
}