tests/testthat/test_deephit.R

In RaphaelS1/survivalmodels: Models for Survival Analysis

skip_if_no_pycox()

set.seed(1)
np <- reticulate::import("numpy")
np$random$seed(1L)
torch <- reticulate::import("torch")
torch$manual_seed(1L)

test_that("silent", {
  expect_silent({fit <- deephit(Surv(time, status) ~ ., data = rats[1:50, ], verbose = FALSE,
                                frac = 0.3)})
  expect_silent({predict(fit, newdata = rats[51:100, ])})
})

test_that("custom", {
  expect_silent(deephit(Surv(time, status) ~ ., data = rats[1:10, ],
                       custom_net = build_pytorch_net(3L, 10L),
                       verbose = FALSE))
})

test_that("auto sanity", {
  sanity_check(model = "deephit",
               pars = list(frac = 0.3, activation = "relu", num_nodes = c(2L, 2L),
                           dropout = 0.1, early_stopping = TRUE, epochs = 100L, batch_size = 32L))
})