R/fn_impute.R

In ready4-dev/youthvars: Describe and Validate Youth Mental Health Economic Datasets

#' Impute adult Assessment of Quality of Life Six Dimension items tibble
#' @description impute_adult_aqol6d_items_tb() is an Impute function that imputes data. Specifically, this function implements an algorithm to impute adult assessment of quality of life six dimension items tibble. The function returns Assessment of Quality of Life Six Dimension items (a tibble).
#' @param aqol6d_items_tb Assessment of Quality of Life Six Dimension items (a tibble)
#' @param domain_items_ls Domain items (a list)
#' @return Assessment of Quality of Life Six Dimension items (a tibble)
#' @rdname impute_adult_aqol6d_items_tb
#' @export 
#' @importFrom lifecycle deprecate_soft
#' @importFrom purrr reduce map2_dbl
#' @importFrom dplyr mutate select mutate_at vars
#' @importFrom rlang sym
#' @keywords internal
impute_adult_aqol6d_items_tb <- function (aqol6d_items_tb, domain_items_ls) 
{
    lifecycle::deprecate_soft("0.0.0.9078", "youthvars::impute_adult_aqol6d_items_tb()", 
        "scorz::impute_adult_aqol6d_items_tb()")
    aqol6d_items_tb <- 1:length(domain_items_ls) %>% purrr::reduce(.init = aqol6d_items_tb, 
        ~{
            idx_int <- .y
            tb <- .x %>% dplyr::mutate(`:=`(!!rlang::sym(paste0("D", 
                names(domain_items_ls)[idx_int], "miss")), rowMeans(dplyr::select(., 
                domain_items_ls[[idx_int]]), na.rm = TRUE)))
            tb %>% dplyr::mutate_at(dplyr::vars(domain_items_ls[[idx_int]]), 
                ~.x %>% purrr::map2_dbl(!!rlang::sym(paste0("D", 
                  names(domain_items_ls)[idx_int], "miss")), 
                  ~{
                    ifelse(is.na(.x), .y, .x)
                  }))
            tb
        })
    return(aqol6d_items_tb)
}
#' Impute unscored adolescent Assessment of Quality of Life Six Dimension dataset
#' @description impute_unscrd_adol_aqol6d_ds() is an Impute function that imputes data. Specifically, this function implements an algorithm to impute unscored adolescent assessment of quality of life six dimension dataset. The function returns Imputed unscored Assessment of Quality of Life dataset tibble (a tibble).
#' @param unscrd_aqol_ds_tb Unscored Assessment of Quality of Life dataset (a tibble)
#' @return Imputed unscored Assessment of Quality of Life dataset tibble (a tibble)
#' @rdname impute_unscrd_adol_aqol6d_ds
#' @export 
#' @importFrom lifecycle deprecate_soft
#' @importFrom dplyr mutate select filter
#' @importFrom mice mice complete
#' @keywords internal
impute_unscrd_adol_aqol6d_ds <- function (unscrd_aqol_ds_tb) 
{
    lifecycle::deprecate_soft("0.0.0.9078", "youthvars::impute_unscrd_adol_aqol6d_ds()", 
        "scorz::impute_unscrd_adol_aqol6d_ds()")
    unscrd_aqol_ds_tb <- unscrd_aqol_ds_tb %>% dplyr::mutate(missing = rowSums(is.na(dplyr::select(., 
        paste0("Q", c(1:10))))))
    aqol_cases_to_imp_tb <- unscrd_aqol_ds_tb %>% dplyr::filter(missing < 
        10) %>% dplyr::select(-missing)
    aqol_cases_not_to_imp_tb <- unscrd_aqol_ds_tb %>% dplyr::filter(missing >= 
        10) %>% dplyr::select(-missing)
    imputed_aqol_tb <- mice::mice(aqol_cases_to_imp_tb, m = 1, 
        maxit = 50, meth = "pmm", seed = 1234)
    aqol_cases_to_imp_tb <- mice::complete(imputed_aqol_tb, "long") %>% 
        dplyr::select(-.imp, -.id)
    imputed_unscrd_aqol_ds_tb_tb <- data.frame(rbind(aqol_cases_to_imp_tb, 
        aqol_cases_not_to_imp_tb))
    return(imputed_unscrd_aqol_ds_tb_tb)
}