test_that("paramtest surv.deephit train", {
learner = lrn("surv.deephit")
fun_list = list(survivalmodels::deephit, survivalmodels:::get_pycox_optim) # nolint
exclude = c(
"formula", # unused
"data", # handled internally
"reverse", # unused
"time_variable", # handled internally
"status_variable", # handled internally
"x", # unused
"y", # unused
"net" # handled internally
)
paramtest = run_paramtest(learner, fun_list, exclude, tag = "train")
expect_paramtest(paramtest)
})
test_that("paramtest surv.deephit predict", {
learner = lrn("surv.deephit")
fun = survivalmodels:::predict.pycox # nolint
exclude = c(
"object", # handled internally
"newdata", # handled internally
"type", # handled internally
"distr6" # handled internally
)
paramtest = run_paramtest(learner, fun, exclude, tag = "predict")
expect_paramtest(paramtest)
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.