coalesce_one_hot: Coalesce one-hot encoding back to a single variable

In rMIDAS: Multiple Imputation with Denoising Autoencoders

Coalesce one-hot encoding back to a single variable

Description

Helper function to reverse one-hot encoding post-imputation.

Usage

coalesce_one_hot(X, var_name, fast = TRUE)

Arguments

X	A data.frame, data.table or matrix, for a single variable
var_name	A character string, with the original variable label
fast	Boolean, indicating whether to choose category with highest predicted probability (TRUE), or use predicted probabilities as weights in draw from random distribution

Value

A vector of length equal to nrow(X), containing categorical labels corresponding to the columns of X

rMIDAS documentation built on Oct. 11, 2023, 5:14 p.m.