tests/testthat/test_surv_pchazard.R

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

test_that("autotest", {
  learner = LearnerSurvPCHazard$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.pchazard", custom_net = net)
  task = tgen("simsurv")$generate(200)
  expect_silent(learner$train(task))
  expect_prediction_surv(learner$predict(task))
  expect_equal(learner$model$model$net, net)
})

test_that("interpolate", {
  learner = lrn("surv.pchazard", sub = 10L)
  task = tgen("simsurv")$generate(200)
  expect_silent(learner$train(task))
  expect_prediction_surv(learner$predict(task))
  expect_equal(learner$model$model$sub, 10L)
})

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