R/sample_subset_pop_dectree.R

In n8thangreen/ltbiScreenLite: ltbiScreenLite

#' sample_subset_pop_dectree
#'
#' Iteratively randomly samples probabilities
#' and then calculates subset sizes
#'
#' @param osNode data.tree object
#' @param n Sample size
#' @param sample_p Random sample TRUE/FALSE; default: TRUE
#'
#' @return matrix
#' @export
#'
#' @examples
#'
sample_subset_pop_dectree <- function(osNode,
                                      n = 1,
                                      sample_p = TRUE) {

  # initiate
  out1 <- subset_pop_dectree(osNode = osNode)

  out <-
    matrix(NA, nrow = n, ncol = ncol(out1)) %>%
    `colnames<-`(colnames(out1))

  for (i in seq_len(n)) {

    osNode$Set(path_probs =
                 calc_pathway_probs(osNode,
                                    sample_p = sample_p))
    out[i, ] <-
      subset_pop_dectree(osNode = osNode) %>%
      as.matrix()
  }

  out
}