tests/testthat/test_surv_deephit.R

In mlr3learners/mlr3learners.pycox: Adds Learners from Python Package pycox for mlr3

test_that("autotest", {
  learner = LearnerSurvDeephit$new()
  expect_learner(learner)
  result = run_autotest(learner, check_replicable = FALSE)
  expect_true(result, info = result$error)
})

test_that("custom net", {
  net = build_pytorch_net(3L, 10L, nodes = c(2, 4, 8, 32), activation = "elu")
  learner = lrn("surv.deephit", custom_net = net)
  task = tgen("simsurv")$generate(50)
  expect_silent(learner$train(task))
  expect_prediction_surv(learner$predict(task))
  expect_equal(learner$model$model$net, net)
})

test_that("interpolate", {
  learner = lrn("surv.deephit", interpolate = TRUE)
  task = tgen("simsurv")$generate(50)
  expect_silent(learner$train(task))
  expect_prediction_surv(learner$predict(task))
})

test_that("validation set", {
  learner = lrn("surv.deephit", frac = 0.2)
  task = tgen("simsurv")$generate(50)
  expect_silent(learner$train(task))
  expect_prediction_surv(learner$predict(task))
})