Nothing
context("Unit tests for IRM, binary outcome")
library("mlr3learners")
lgr::get_logger("mlr3")$set_threshold("warn")
on_cran = !identical(Sys.getenv("NOT_CRAN"), "true")
if (on_cran) {
test_cases = expand.grid(
learner = "rpart",
dml_procedure = "dml1",
score = "ATTE",
trimming_threshold = 0,
stringsAsFactors = FALSE
)
test_cases[".test_name"] = apply(test_cases, 1, paste, collapse = "_")
} else {
test_cases = expand.grid(
learner = "cv_glmnet",
dml_procedure = c("dml1", "dml2"),
score = c("ATE", "ATTE"),
trimming_threshold = 0,
stringsAsFactors = FALSE
)
test_cases[".test_name"] = apply(test_cases, 1, paste, collapse = "_")
}
patrick::with_parameters_test_that("Unit tests for IRM:",
.cases = test_cases,
{
learner_pars = get_default_mlmethod_irm_binary(learner)
n_rep_boot = 498
set.seed(3141)
irm_hat = dml_irm(data_irm_binary$df,
y = "y", d = "d",
n_folds = 5,
ml_g = learner_pars$ml_g$clone(), ml_m = learner_pars$ml_m$clone(),
dml_procedure = dml_procedure, score = score
)
theta = irm_hat$coef
se = irm_hat$se
boot_theta = bootstrap_irm(irm_hat$thetas, irm_hat$ses,
data_irm_binary$df,
y = "y", d = "d",
n_folds = 5, smpls = irm_hat$smpls,
all_preds = irm_hat$all_preds,
score = score,
bootstrap = "normal", n_rep_boot = n_rep_boot
)$boot_coef
set.seed(3141)
double_mlirm_obj = DoubleMLIRM$new(
data = data_irm_binary$dml_data,
n_folds = 5,
ml_g = learner_pars$ml_g$clone(),
ml_m = learner_pars$ml_m$clone(),
dml_procedure = dml_procedure,
score = score,
trimming_threshold = trimming_threshold
)
double_mlirm_obj$fit()
theta_obj = double_mlirm_obj$coef
se_obj = double_mlirm_obj$se
# bootstrap
double_mlirm_obj$bootstrap(method = "normal", n_rep = n_rep_boot)
boot_theta_obj = double_mlirm_obj$boot_coef
# at the moment the object result comes without a name
expect_equal(theta, theta_obj, tolerance = 1e-8)
expect_equal(se, se_obj, tolerance = 1e-8)
expect_equal(as.vector(boot_theta), as.vector(boot_theta_obj), tolerance = 1e-8)
}
)
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.