skip_if_not_installed("mlr3")
test_that("mlr3 learner description can be printed", {
task = mlr3::tsk("pima")
learner = mlr3::lrn("classif.rpart")
learner$train(task)
v <- vetiver_model(learner, "rpart_pima")
expect_snapshot(v)
})
test_that("mlr3 learners can be pinned", {
task = mlr3::tsk("pima")
learner = mlr3::lrn("classif.rpart")
learner$train(task)
v <- vetiver_model(learner, "rpart_pima")
b <- board_temp()
vetiver_pin_write(b, v)
pinned <- pin_read(b, "rpart_pima")
expect_length(pinned, 2)
expect_equal(class(pinned$model)[1], "LearnerClassifRpart")
expect_equal(nrow(pinned$prototype), 0)
expect_equal(names(pinned$prototype), task$feature_names)
expect_equal(pin_meta(b, "rpart_pima")$user$required_pkgs,
c("mlr3", "rpart"))
})
test_that("learners from mlr3learners can be pinned", {
skip_if_not_installed("mlr3learners")
task = mlr3::tsk("spam")
learner = mlr3learners::LearnerClassifXgboost$new()
learner$train(task)
v <- vetiver_model(learner, "logreg_spam")
b <- board_temp()
vetiver_pin_write(b, v)
pinned <- pin_read(b, "logreg_spam")
expect_length(pinned, 2)
expect_equal(class(pinned$model)[1], "LearnerClassifXgboost")
expect_equal(nrow(pinned$prototype), 0)
expect_equal(names(pinned$prototype), task$feature_names)
expect_equal(pin_meta(b, "logreg_spam")$user$required_pkgs,
c("mlr3", "mlr3learners", "xgboost"))
})
test_that("create plumber.R for mlr3", {
skip_on_cran()
task = mlr3::tsk("pima")
learner = mlr3::lrn("classif.rpart")
learner$train(task)
v <- vetiver_model(learner, "rpart_pima")
b <- board_folder(path = tmp_dir)
vetiver_pin_write(b, v)
tmp <- tempfile()
vetiver_write_plumber(b, "rpart_pima", file = tmp)
expect_snapshot(
cat(readr::read_lines(tmp), sep = "\n"),
transform = redact_vetiver
)
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.