helper methods

Author: topepo

NAMESPACE

@@ -2,6 +2,7 @@
 
 S3method(.censoring_weights_graf,default)
 S3method(.censoring_weights_graf,model_fit)
+S3method(as.matrix,quantile_pred)
 S3method(as_tibble,quantile_pred)
 S3method(augment,model_fit)
 S3method(autoplot,glmnet)
@@ -232,6 +233,7 @@ export(extract_fit_engine)
 export(extract_fit_time)
 export(extract_parameter_dials)
 export(extract_parameter_set_dials)
+export(extract_quantile_levels)
 export(extract_spec_parsnip)
 export(find_engine_files)
 export(fit)

R/aaa_quantiles.R

@@ -64,6 +64,9 @@ new_quantile_pred <- function(values = list(), quantile_levels = double()) {
 #'
 #' # tidy format
 #' as_tibble(v)
+#'
+#' # matrix format
+#' as.matrix(v)
 quantile_pred <- function(values, quantile_levels = double()) {
   check_quantile_pred_inputs(values, quantile_levels)
   quantile_levels <- vctrs::vec_cast(quantile_levels, double())
@@ -178,3 +181,18 @@ as_tibble.quantile_pred <-
       .row = rep(1:n_samp, each = n_quant)
     )
   }
+
+#' @export
+as.matrix.quantile_pred <- function(x, ...) {
+  num_samp <- length(x)
+  matrix(unlist(x), nrow = num_samp)
+}
+
+#' @export
+#' @rdname quantile_pred
+extract_quantile_levels <- function(x) {
+  if ( !inherits(x, "quantile_pred") ) {
+    cli::cli_abort("{.arg x} should have class {.val quantile_pred}.")
+  }
+  attr(x, "quantile_levels")
+}