tests/testcases/Thrf_test.R

In soerenkuenzel/causalToolbox: Toolbox for Causal Inference with emphasize on Heterogeneous Treatment Effect Estimator

devtools::load_all()
# Example 1: Use Iris dataset
set.seed(1423614230)

feat <- iris[, -1]
tr <- rbinom(nrow(iris), 1, .5)
yobs <- iris[, 1]

tl <- T_RF(
  feat = feat,
  tr = tr,
  yobs = yobs,
  nthread = 1
)

EstimateCate(tl, feat)


# Example 2: large n
set.seed(400189)
n <- 10000
d <- 12

feat <- as.data.frame(matrix(rnorm(n*d), nrow = n))
tr <- rbinom(n, 1, feat$V1/ 20 +.5)
yobs <- feat$V1  + feat$V2 + tr * feat$V3

# this takes approximately 2 min to run.
tl <- T_RF(
  feat = feat,
  tr = tr,
  yobs = yobs
)