SAGE with marginal sampling (features are marginalized independently). This is the standard SAGE implementation.
Super classes
xplainfi::FeatureImportanceMethod -> xplainfi::SAGE -> MarginalSAGE
Methods
Method new()
Creates a new instance of the MarginalSAGE class.
Usage
MarginalSAGE$new(
task,
learner,
measure,
resampling = NULL,
features = NULL,
n_permutations = 10L,
batch_size = 5000L,
n_samples = 100L,
early_stopping = FALSE,
se_threshold = 0.01,
min_permutations = 3L,
check_interval = 1L
)Arguments
task, learner, measure, resampling, features, n_permutations, batch_size, n_samples, early_stopping, se_threshold, min_permutations, check_intervalPassed to SAGE.
Examples
library(mlr3)
task = tgen("friedman1")$generate(n = 100)
sage = MarginalSAGE$new(
task = task,
learner = lrn("regr.ranger", num.trees = 50),
measure = msr("regr.mse"),
n_permutations = 3L,
n_samples = 20
)
#> ℹ No <Resampling> provided
#> Using `resampling = rsmp("holdout")` with default `ratio = 0.67`.
sage$compute()