From b436d0c8bd6f312ecdcbab9c2a3d2fdb863b955c Mon Sep 17 00:00:00 2001 From: Tom Date: Wed, 15 Sep 2021 16:34:38 -0500 Subject: [PATCH] Modify candsetmaxpro to sample with replacement if candmat has fewer than npoints rows --- R/ModelFrameProcessing.R | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/R/ModelFrameProcessing.R b/R/ModelFrameProcessing.R index a54343f..a0d6c83 100644 --- a/R/ModelFrameProcessing.R +++ b/R/ModelFrameProcessing.R @@ -510,7 +510,10 @@ candsetmaxpro <- function(candmat, npoints = NULL, groupid = NULL){ if(is.null(groupid)){ - designvec <- sample(c(1:nrow(D0)), npoints) + initial_replacement = FALSE + if(nrow(D0) < npoints){initial_replacement = TRUE} + + designvec <- sample(c(1:nrow(D0)), npoints, replace = initial_replacement) }else{ designvec <- sapply(c(1:npoints), function(x) sample(which(groupid == x), size = 1))