https://github.com/tom-n-walker/uphill-plants-soil-carbon
Revision 84beafabeca10c7be201c8070b5bc8ef12e3255b authored by Tom Walker on 06 September 2021, 07:47:08 UTC, committed by Tom Walker on 06 September 2021, 07:47:08 UTC
1 parent bf80ebd
Tip revision: 84beafabeca10c7be201c8070b5bc8ef12e3255b authored by Tom Walker on 06 September 2021, 07:47:08 UTC
Added data processing scripts; edited .gitignore.
Added data processing scripts; edited .gitignore.
Tip revision: 84beafa
apply_mice.R
################################################################################
#### Project: General functionality
#### Title: Function | Small function | Apply MICE imputation
#### Author: Tom Walker (thomas.walker@usys.ethz.ch)
#### Date: 15 July 2020
#### ---------------------------------------------------------------------------
apply_mice <- function(matrix, iter){
# get column names
cols <- colnames(matrix)
colnames(matrix) <- paste0("C", 1:ncol(matrix))
# apply mice imputation
imp <- mice(
matrix,
iter
)
# generate list and add each imputation round to a bin
imp_list <- list()
for(i in 1:iter){
imp_list[[i]] <- as.matrix(
complete(imp, i)
)
}
# take means of these iterations
out <- apply(
simplify2array(imp_list),
1:2,
mean,
na.rm = T
) %>% as.data.frame
colnames(out) <- cols
# return
return(out)
}
Computing file changes ...