R/stan_BF_data.R

In lgaborini/rstanBF: rstanBF computes the Bayes Factor for data in specified two-level hierarchical models

#' Convert the output of rsamplestudy to rstanBF data parameter
#'
#' @param list_pop the output of a `rsamplestudy`-generated population
#' @param list_samples the output of [rsamplestudy::make_dataset_splits()]
#' @param col_source the name of the source column (default: 'source')
#' @return a list containing `mtx`, `idx.ref`, `idx.quest`
#' @export
stanBF_prepare_rsamplestudy_data <- function(list_pop, list_samples, col_source = 'source') {

   stopifnot(is.list(list_pop))
   stopifnot(all(c('df_pop') %in% names(list_pop)))
   stopifnot(col_source %in% colnames(list_pop$df_pop))

   stopifnot(is.list(list_samples))
   stopifnot(all(c('idx_reference', 'idx_questioned') %in% names(list_samples)))

   mtx_samples <- dplyr::select(list_pop$df_pop, -col_source) %>% as.matrix()

   list_data <- list(
      mtx = mtx_samples,
      idx.ref = list_samples$idx_reference,
      idx.quest = list_samples$idx_questioned
   )
   list_data
}