tests/testthat/test_TaskClassif_melanoma.R

In mlr3torch: Deep Learning with 'mlr3'

skip_on_cran()
skip_if_not_installed("curl")

test_that("melanoma task works", {
  withr::local_options(mlr3torch.cache = TRUE)
  task = tsk("melanoma")

  expect_equal(task$id, "melanoma")
  expect_equal(task$label, "Melanoma Classification")
  expect_equal(task$feature_names, c("sex", "anatom_site_general_challenge", "age_approx", "image"))
  expect_equal(task$target_names, "outcome")
  expect_equal(task$man, "mlr3torch::mlr_tasks_melanoma")
  expect_equal(task$properties, c("twoclass", "groups"))

  task$data()
  x = materialize(task$data(1, "image")[[1]])[[1L]]
  expect_class(x, "torch_tensor")
  expect_equal(x$shape, c(3, 128, 128))

  expect_true("data.rds" %in% list.files(file.path(get_cache_dir(), "datasets", "melanoma")))
  expect_equal(task$backend$nrow, 32701 + 10982)
  expect_equal(task$backend$ncol, 11)
  expect_equal(task$ncol, 5)
})