ebalance
ObjectsGenerates balance statistics for output objects from ebal.
# S3 method for ebalance
bal.tab(
x,
formula = NULL,
data = NULL,
treat = NULL,
covs = NULL,
stats,
int = FALSE,
poly = 1,
distance = NULL,
addl = NULL,
continuous,
binary,
s.d.denom,
thresholds = NULL,
weights = NULL,
cluster = NULL,
imp = NULL,
pairwise = TRUE,
s.weights = NULL,
abs = FALSE,
subset = NULL,
quick = TRUE,
...
)
If clusters and imputations are not specified, an object of class "bal.tab"
containing balance summaries for the given object. See bal.tab()
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 class-bal.tab.cluster
for details.
an ebalance
object (the output of a call to ebal::ebalance()
or ebal::ebalance.trim()
).
a formula
with the treatment variable as the response and the covariates for which balance is to be assessed as the predictors. All named variables must be in data
. See Details.
a data frame containing variables named in formula
, if supplied, and other arguments.
a vector of treatment statuses. See Details.
a data frame of covariate values for which to check balance. See Details.
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 in the argument to data
, if specified. For longitudinal treatments, can be a list of allowable arguments, one for each time point.
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 in the arguments to the input object, covs
, and data
, if specified. For longitudinal treatments, can be a list of allowable arguments, one for each time point.
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()
.
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()
.
character
; how the denominator for standardized mean differences should be calculated, if requested. See col_w_smd()
for allowable options. Abbreviations allowed. If not specified, will be set to "treated"
.
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. When specified, extra columns are inserted into the Balance table describing whether the requested balance statistics exceeded the threshold or not. 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 are added to the output.
a vector, list, or data.frame
containing weights for each unit, or a string containing the names of the weights variables in data
, or an object with a get.w()
method or a list thereof. The weights can be, e.g., inverse probability weights or matching weights resulting from a matching algorithm.
either a vector containing cluster membership for each unit or a string containing the name of the cluster membership variable in data
or the input object. See class-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 input object. See class-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.
Optional; either a vector containing sampling weights for each unit or a string containing the name of the sampling weight variable in data
. These function like regular weights except that both the adjusted and unadjusted samples will be weighted according to these weights if weights are used.
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 have length equal to the number of units. 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.
for some input types, other arguments that are required or allowed. Otherwise, further arguments to control display of output. See display options for details.
bal.tab()
generates a list of balance summaries for the object given. The input to bal.tab.ebalance()
must include either both formula
and data
or both covs
and treat
.
bal.tab()
for details of calculations.
if (FALSE) { # requireNamespace("ebal", quietly = TRUE)
data("lalonde", package = "cobalt")
covs <- subset(lalonde, select = -c(re78, treat))
covs0 <- splitfactor(covs)
e.out <- ebal::ebalance(lalonde$treat, covs0)
## Using formula and data
bal.tab(e.out, formula = treat ~ age + educ + race +
married + nodegree + re74 + re75,
data = lalonde)
## Using treat and covs
bal.tab(e.out, treat = lalonde$treat, covs = covs)
}
Run the code above in your browser using DataLab