tests/testthat/_problems/test-WVIM-340.R

In xplainfi: Feature Importance Methods for Global Explanations

# Extracted from test-WVIM.R:340

# setup ------------------------------------------------------------------------
library(testthat)
test_env <- simulate_test_env(package = "xplainfi", path = "..")
attach(test_env, warn.conflicts = FALSE)

# test -------------------------------------------------------------------------
task = tgen("friedman1")$generate(n = 200)
wvim = WVIM$new(
		task = task,
		learner = lrn("regr.rpart"),
		measure = msr("regr.mse"),
		resampling = rsmp("holdout"),
		features = task$feature_names[1:3],
		direction = "leave-out",
		n_repeats = 1L
	)
wvim$compute()
imp = wvim$importance(ci_method = "lei")