From 44842c67da2f73dd6236df26db2dc0195381a32f Mon Sep 17 00:00:00 2001 From: MM Date: Thu, 15 Mar 2018 11:14:23 +0100 Subject: [PATCH] pass prob to holdout_frac_() --- R/holdout.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/holdout.R b/R/holdout.R index e9d06bf..1cae4f1 100644 --- a/R/holdout.R +++ b/R/holdout.R @@ -47,7 +47,7 @@ holdout_frac.data.frame <- function(data, size = 0.3, K = 1L, shuffle = TRUE, assert_that(is.flag(shuffle)) assert_that(is.null(prob) || (is.numeric(prob) && length(prob) %in% c(1, nrow(data)))) - res <- holdout_frac_(nrow(data), size = size, K = K, shuffle = shuffle) + res <- holdout_frac_(nrow(data), size = size, K = K, shuffle = shuffle, prob=prob) to_crossv_df(res, data) }