WeightIt
ObjectsGenerates balance statistics for weightit
and weightitMSM
objects from WeightIt.
# S3 method for weightit
bal.tab(x,
stats,
int = FALSE,
poly = 1,
distance = NULL,
addl = NULL,
data = NULL,
continuous,
binary,
s.d.denom,
thresholds = NULL,
weights = NULL,
cluster = NULL,
imp = NULL,
pairwise = TRUE,
abs = FALSE,
subset = NULL,
quick = TRUE,
...)# S3 method for weightitMSM
bal.tab(x,
stats,
int = FALSE,
poly = 1,
distance.list = NULL,
addl.list = NULL,
data = NULL,
continuous,
binary,
s.d.denom,
thresholds = NULL,
weights = NULL,
cluster = NULL,
imp = NULL,
pairwise = TRUE,
abs = FALSE,
subset = NULL,
quick = TRUE,
...)
For point treatments, if clusters and imputations are not specified, an object of class "bal.tab"
containing balance summaries for the weightit
object. See bal.tab()
for details.
If imputations are specified, an object of class "bal.tab.imp"
containing balance summaries for each imputation and a summary of balance across imputations. See bal.tab.imp
for details.
If weightit()
is used with multi-category treatments, an object of class "bal.tab.multi"
containing balance summaries for each pairwise treatment comparison. See bal.tab.multi
for details.
If weightitMSM()
is used for longitudinal treatments, an object of class "bal.tab.msm"
containing balance summaries for each time period. See bal.tab.msm
for details.
If clusters are specified, an object of class "bal.tab.cluster"
containing balance summaries within each cluster and a summary of balance across clusters. See bal.tab.cluster
for details.
a weightit
or weightitMSM
object; the output of a call to WeightIt::weightit()
or WeightIt::weightitMSM()
.
character
; which statistic(s) should be reported. See stats
for allowable options. For binary and multi-category treatments, "mean.diffs"
(i.e., mean differences) is the default. For continuous treatments, "correlations"
(i.e., treatment-covariate Pearson correlations) is the default. Multiple options are allowed.
logical
or numeric
; whether or not to include 2-way interactions of covariates included in covs
and in addl
. If numeric
, will be passed to poly
as well.
numeric
; the highest polynomial of each continuous covariate to display. For example, if 2, squares of each continuous covariate will be displayed (in addition to the covariate itself); if 3, squares and cubes of each continuous covariate will be displayed, etc. If 1, the default, only the base covariate will be displayed. If int
is numeric, poly
will take on the value of int
.
an optional formula or data frame containing distance values (e.g., propensity scores) or a character vector containing their names. If a formula or variable names are specified, bal.tab()
will look first in the argument to data
, if specified, and next in the weightit
or weightitMSM
object. Note that propensity scores generated by weightit()
and weightitMSM()
are automatically included and named "prop.score". For weightitMSM
objects, can be a list of distance values as described above, with one list entry per time period.
an optional formula or data frame containing additional covariates for which to present balance or a character vector containing their names. If a formula or variable names are specified, bal.tab()
will look first in the argument to data
, if specified, and next in the weightit
or weightitMSM
object. For weightitMSM
objects, can be a list of additional covariate values as described above, with one list entry per time period.
an optional data frame containing variables that might be named in arguments to distance
, addl
, cluster
, and imp
. Can also be mids
object, the output of a call to mice()
from the mice package, containing multiply imputed data sets. In this case, imp
is automatically supplied using the imputation variable created from processing the mids
object.
whether mean differences for continuous variables should be standardized ("std"
) or raw ("raw"
). Default "std"
. Abbreviations allowed. This option can be set globally using set.cobalt.options()
. This argument is used to set std
in col_w_smd()
and col_w_cov()
.
whether mean differences for binary variables (i.e., difference in proportion) should be standardized ("std"
) or raw ("raw"
). Default "raw"
. Abbreviations allowed. This option can be set globally using set.cobalt.options()
. This argument is used to set std
in col_w_smd()
and col_w_cov()
.
character
; how the denominator for standardized mean differences should be calculated, if requested. See col_w_smd()
for allowable options. Can also be the name of the treatment group or "focal"
if focal
is specified in the original call to weightit
. If not specified, bal.tab()
will figure out which one is best based on the estimand of the weightit
object: if ATT, "treated"
; if ATC, "control"
; otherwise "pooled"
. Abbreviations allowed.
a named vector of balance thresholds, where the name corresponds to the statistic (i.e., in stats
) that the threshold applies to. For example, to request thresholds on mean differences and variance ratios, one can set thresholds = c(m = .05, v = 2)
. Requesting a threshold automatically requests the display of that statistic. See Details.
a named list containing additional weights on which to assess balance. Each entry can be a vector of weights, the name of a variable in data
that contains weights, or an object with a get.w()
method.
either a vector containing cluster membership for each unit or a string containing the name of the cluster membership variable in data
or the weightit
object. See bal.tab.cluster
for details.
either a vector containing imputation indices for each unit or a string containing the name of the imputation index variable in data
or the original data set used in the call to weightit()
. See bal.tab.imp
for details. Not necessary if data
is a mids
object.
whether balance should be computed for pairs of treatments or for each treatment against all groups combined. See bal.tab.multi
for details. This can also be used with a binary treatment to assess balance with respect to the full sample.
logical
; whether displayed balance statistics should be in absolute value or not.
a logical
or numeric
vector denoting whether each observation should be included or which observations should be included. If logical
, it should be the same length as the variables in the original call to weightit()
or weightitMSM()
. NA
s will be treated as FALSE
. This can be used as an alternative to cluster
to examine balance on subsets of the data.
logical
; if TRUE
, will not compute any values that will not be displayed. Set to FALSE
if computed values not displayed will be used later.
further arguments to control display of output. See display options for details.
Noah Greifer
bal.tab.weightit()
generates a list of balance summaries for the weightit
object given.
The threshold
argument controls whether extra columns should be inserted into the Balance table describing whether the balance statistics in question exceeded or were within the threshold. Including these thresholds also creates summary tables tallying the number of variables that exceeded and were within the threshold and displaying the variables with the greatest imbalance on that balance measure.
bal.tab()
for details of calculations.
if (all(sapply(c("WeightIt", "twang"), requireNamespace, quietly = TRUE))) {
library(WeightIt)
data("lalonde", package = "cobalt")
## Basic propensity score weighting
w.out1 <- weightit(treat ~ age + educ + race +
married + nodegree + re74 + re75,
data = lalonde, method = "ps")
bal.tab(w.out1, un = TRUE, m.threshold = .1,
v.threshold = 2)
## Weighting with a multi-category treatment
w.out2 <- weightit(race ~ age + educ + married +
nodegree + re74 + re75,
data = lalonde, method = "ps",
estimand = "ATE", use.mlogit = FALSE)
bal.tab(w.out2, un = TRUE)
bal.tab(w.out2, un = TRUE, pairwise = FALSE)
## IPW for longitudinal treatments
data("iptwExWide", package = "twang")
wmsm.out <- weightitMSM(list(tx1 ~ use0 + gender,
tx2 ~ use0 + gender + use1 + tx1,
tx3 ~ use0 + gender + use1 + tx1 + use2 + tx2),
data = iptwExWide,
stabilize = TRUE)
bal.tab(wmsm.out)
}
Run the code above in your browser using DataLab