plot_2num1fac {mixgb} | R Documentation |
Plot observed values versus m sets of imputed values for two specified numeric variables and a factor using ggplot2.
plot_2num1fac(
imputation.list,
var.x,
var.y,
con.fac,
original.data,
true.data = NULL,
color.pal = NULL,
shape = FALSE
)
imputation.list |
A list of |
var.x |
A numeric variable on the x-axis |
var.y |
A numeric variable on the y-axis |
con.fac |
The name of a factor to condition on |
original.data |
The original data with missing data |
true.data |
The true data without missing values. This is generally unknown in practice. If the true data is known (e.g., in cases where it is generated by simulation), it can be specified in this argument. The output will then have an extra panel called |
color.pal |
A vector of hex color codes for the observed and m sets of imputed values panels. The vector should be of length |
shape |
Whether to plot shapes for different types of missing values. By default, this is set to FALSE to speed up plotting. We only recommend using 'shape = TRUE' for small datasets. |
Scatter plots
# obtain m multiply datasets
params <- list(max_depth = 3, subsample = 0.8, nthread = 2)
imputed.data <- mixgb(data = nhanes3, m = 3, xgb.params = params, nrounds = 30)
# plot the multiply imputed values for variables "BMPRECUM" versus "BMPHEAD" conditional on "HSSEX"
plot_2num1fac(
imputation.list = imputed.data, var.x = "BMPHEAD", var.y = "BMPRECUM",
con.fac = "HSSEX", original.data = nhanes3
)