sage_batch_predict: Batch Predict for SAGE

In xplainfi: Feature Importance Methods for Global Explanations

Batch Predict for SAGE

Description

Performs batched prediction on combined data to manage memory usage. Supports both classification (probability predictions) and regression.

Usage

sage_batch_predict(learner, combined_data, task, batch_size, task_type)

Arguments

learner	(Learner) Trained mlr3 learner.
combined_data	(data.table) Data with feature columns to predict on.
task	(Task) mlr3 task object.
batch_size	(integer(1) or NULL) Batch size for predictions. If NULL or if total_rows <= batch_size, processes all data at once.
task_type	(character(1)) Task type, either "classif" or "regr".

Value

For classification: matrix of class probabilities (n_rows x n_classes). For regression: numeric vector of predictions (length n_rows).

xplainfi documentation built on Feb. 27, 2026, 1:08 a.m.