Skip to content

Commit

Permalink
rem gbm
Browse files Browse the repository at this point in the history
  • Loading branch information
jaredhuling committed May 21, 2019
1 parent d911234 commit d0b624c
Showing 1 changed file with 26 additions and 26 deletions.
52 changes: 26 additions & 26 deletions tests/testthat/test-fitsubgroup.R
Original file line number Diff line number Diff line change
Expand Up @@ -780,16 +780,16 @@ test_that("test fit.subgroup for continuous outcomes and various losses", {

expect_is(subgrp.model, "subgroup_fitted")

subgrp.model <- fit.subgroup(x = x, y = y,
trt = trt01,
propensity.func = prop.func,
loss = "sq_loss_gbm",
n.trees = 5,
n.cores = 1)

invisible(capture.output(print(subgrp.model)))
invisible(capture.output(summary(subgrp.model)))
expect_is(subgrp.model, "subgroup_fitted")
# subgrp.model <- fit.subgroup(x = x, y = y,
# trt = trt01,
# propensity.func = prop.func,
# loss = "sq_loss_gbm",
# n.trees = 5,
# n.cores = 1)
#
# invisible(capture.output(print(subgrp.model)))
# invisible(capture.output(summary(subgrp.model)))
# expect_is(subgrp.model, "subgroup_fitted")
}

# subgrp.model <- fit.subgroup(x = x, y = y,
Expand Down Expand Up @@ -1052,16 +1052,16 @@ test_that("test fit.subgroup with augment.func for continuous outcomes and vario
invisible(capture.output(print(subgrp.model)))
invisible(capture.output(summary(subgrp.model)))

subgrp.model <- fit.subgroup(x = x, y = y,
trt = trt01,
propensity.func = prop.func,
loss = "sq_loss_gbm",
n.trees = 5,
n.cores = 1)

invisible(capture.output(print(subgrp.model)))
invisible(capture.output(summary(subgrp.model)))
expect_is(subgrp.model, "subgroup_fitted")
# subgrp.model <- fit.subgroup(x = x, y = y,
# trt = trt01,
# propensity.func = prop.func,
# loss = "sq_loss_gbm",
# n.trees = 5,
# n.cores = 1)
#
# invisible(capture.output(print(subgrp.model)))
# invisible(capture.output(summary(subgrp.model)))
# expect_is(subgrp.model, "subgroup_fitted")

# subgrp.model <- fit.subgroup(x = x, y = y,
# trt = trt01,
Expand Down Expand Up @@ -1534,12 +1534,12 @@ test_that("test fit.subgroup for continuous outcomes and match.id provided", {
loss = "sq_loss_lasso_gam",
foldid = sample(1:5, nrow(x.m), replace = TRUE)))

subgrp.model.m <- fit.subgroup(x = x.m, y = y.m,
trt = trt.m,
match.id = match.id,
loss = "sq_loss_gbm", n.trees = 5, n.cores = 1)

expect_is(subgrp.model.m, "subgroup_fitted")
# subgrp.model.m <- fit.subgroup(x = x.m, y = y.m,
# trt = trt.m,
# match.id = match.id,
# loss = "sq_loss_gbm", n.trees = 5, n.cores = 1)
#
# expect_is(subgrp.model.m, "subgroup_fitted")

# subgrp.model.m <- fit.subgroup(x = x.m, y = y.m,
# trt = trt.m,
Expand Down

0 comments on commit d0b624c

Please sign in to comment.