BART (version 1.6)

stratrs: Perform stratified random sampling to balance outcomes

Description

This function is used to perform stratified random sampling to balance outcomes among the shards.

Usage

stratrs(y, C=floor(length(y)/2000))

Arguments

y

The binary/categorical outcome.

C

The number of shards to break the data set into.

Value

A vector is returned with each element assigned to a shard.

Details

To perform BART with large data sets, random sampling is employed to break the data into C shards. Each shard should be balanced with respect to the outcome. For binary/categorical outcomes, stratified random sampling is employed with this function.

See Also

rs.pbart

Examples

# NOT RUN {
set.seed(12)
x <- rbinom(25000, 1, 0.1)
a <- stratrs(x)
table(a, x)
z <- pmin(rpois(25000, 0.8), 5)
b <- stratrs(z)
table(b, z)
# }