adding pretty method to validation_set

tidymodels · Jan 10, 2024 · d393a8a · d393a8a
1 parent f476210
commit d393a8a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -53,6 +53,7 @@ S3method(pretty,rolling_origin)
 S3method(pretty,sliding_index)
 S3method(pretty,sliding_period)
 S3method(pretty,sliding_window)
+S3method(pretty,validation_set)
 S3method(pretty,validation_split)
 S3method(pretty,vfold_cv)
 S3method(print,apparent)

diff --git a/R/printing.R b/R/printing.R
@@ -63,6 +63,19 @@ pretty.mc_cv <- function(x, ...) {
   res
 }
 
+#' @export
+pretty.validation_set <- function(x, ...) {
+  details <- attributes(x)
+  res <- paste0(
+    "Validation Set (",
+    signif(details$prop[1] / sum(details$prop), 2),
+    "/",
+    signif(details$prop[2] / sum(details$prop), 2),
+    ")"
+  )
+  res
+}
+
 #' @export
 pretty.validation_split <- function(x, ...) {
   details <- attributes(x)