context("sft")
test_that("sft",{
library(parsnip)
rf_no_forest <- randomForest::randomForest(iris[,-5], factor(iris$Species), ntree = 100, keep.forest = FALSE)
set.seed(1234)
rf_w_forest <- randomForest::randomForest(iris[,-5], factor(iris$Species), ntree = 100, keep.forest = TRUE)
ranger_no_forest <- ranger::ranger(factor(iris$Species) ~., iris[,-5], write.forest = FALSE,num.trees = 100)
set.seed(1234)
ranger_w_forest <- ranger::ranger(factor(iris$Species) ~., iris[,-5], write.forest = TRUE, num.trees = 100)
model_parsnip_ranger <-
rand_forest(mode = 'classification') %>% set_engine("ranger", importance = 'impurity') %>%
fit(Species ~ ., iris)
model_parsnip_rf <-
rand_forest(mode = 'classification') %>% set_engine("randomForest") %>%
fit(Species ~ ., iris)
expect_true(is.list(sft(rf_w_forest, 0.1)))
expect_true(is.list(sft(ranger_w_forest, 0.1)))
expect_true(is.list(sft(model_parsnip_ranger, 0.1)))
expect_true(is.list(sft(model_parsnip_rf, 0.1)))
expect_error(is.list(rf_no_forest,0.1))
expect_error(is.list(ranger_no_forest,0.1))
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.