impute_mf: Impute using missForest

In wfmueller29/SLAM: SLAM - Functions for Longitduinal Datasets

Impute using missForest

Description

This function will allow the user to specify factor variables and variables they do not want in the imputation. It will then create a data matrix to impute using missForest. It will add drop columns back after imputation and relabel the factor variables.

Usage

impute_mf(
  data,
  data_true = NULL,
  factors = NULL,
  drop = NULL,
  ntree = 500,
  ...
)

Arguments

data	the dataframe that we would like to impute NAs.
data_true	optional complete dataframe of the dataframe provided as data argument
factors	a character vector of factor variables that we would like specfiy as factors when imputing. These factors variables can be numeric or character variables in the data provided. It is recommended that does not have numeric meaning be included as a factor.
drop	a character vector of variables that will not be included in the imputation. These variables will be added to the outputted dataframe, even though they are not included in the imputation. Variables to include here would be dates as missForest cannot impute dates.
ntree	number of trees to grow in each forest. Default is set to 500.
...	Other arguments to be passed to missForest imputation, besides ntree

Value

A list of output similar to missForest

ximp	a dataframe with no missing values. The resulting dataframe will be of the same size as the original data provided, but with all of the NA's imputed. However, if there are NA's in the drop columns, these values wil not be imputed.
OOBerror	estimated OOB imputation error. For the set of continuous variables in 'xmis' the NRMSE and for the set of categorical variables the proportion of falsely classified entries is returned. See Details for the exact definition of these error measures. If 'variablewise' is set to 'TRUE' then this will be a vector of length 'p' where 'p' is the number of variables and the entries will be the OOB error for each variable separately.
error	true imputation error. This is only available if 'xtrue' was supplied. The error measures are the same as for 'OOBerror'.

Author(s)

William Mueller, Jorge Romero Martinez

wfmueller29/SLAM documentation built on April 5, 2025, 5:09 a.m.