export helper for post-processing of predictions

Author: hfrick

NAMESPACE

@@ -113,6 +113,7 @@ export(.dat)
 export(.facts)
 export(.lvls)
 export(.obs)
+export(.organize_glmnet_pred)
 export(.preds)
 export(.x)
 export(.y)

R/linear_reg.R

@@ -192,7 +192,18 @@ check_args.linear_reg <- function(object) {
 
 # ------------------------------------------------------------------------------
 
-organize_glmnet_pred <- function(x, object) {
+#' Organize glmnet predictions
+#'
+#' This function is for developer use and organizes predictions from glmnet
+#' models.
+#'
+#' @param x Predictions as returned by the `predict()` method for glmnet models.
+#' @param object An object of class `model_fit`.
+#'
+#' @rdname glmnet_helpers_prediction
+#' @keywords internal
+#' @export
+.organize_glmnet_pred <- function(x, object) {
   if (ncol(x) == 1) {
     res <- x[, 1]
     res <- unname(res)

R/linear_reg_data.R

@@ -163,7 +163,7 @@ set_pred(
   type = "numeric",
   value = list(
     pre = NULL,
-    post = organize_glmnet_pred,
+    post = .organize_glmnet_pred,
     func = c(fun = "predict"),
     args =
       list(