R/extract.R

In simongrund1/robosynth: Robust Synthetic Data Generation with Data-Augmented Multiple Imputation

# // Extract list of synthetic data sets

extract <- function(object, drop = TRUE) {

  syn_list <- object$syn

  N <- nrow(syn_list[[1]])
  m <- length(syn_list)

  # drop masked variables
  if (drop) {
    ind_mask <- sapply(object$models, inherits, "robosynth.masking.model")
    var_mask <- names(object$models)[ind_mask]
    vars <- !(colnames(syn_list[[1]]) %in% var_mask)
    for(jj in 1L:m) {
      syn_list[[jj]] <- syn_list[[jj]][, vars, drop = FALSE]
    }
  }

  class(syn_list) <- c("robosynth.list", "list")
  return(syn_list)

}