get_ensemble_predictions: Get Training Predictions from an Ensemble Model

In e2tree: Explainable Ensemble Trees

Get Training Predictions from an Ensemble Model

Description

Returns a numeric vector of length n_obs with the ensemble's prediction for every training observation. For models that store out-of-bag (OOB) predictions (randomForest, ranger) the stored OOB vector is returned; for other models in-sample predictions are computed from the training data.

Usage

get_ensemble_predictions(ensemble, data, type)

Arguments

ensemble	A trained ensemble model.
data	The training data.frame that was used to fit the model.
type	Character: "classification" or "regression".

Value

Numeric vector of length nrow(data).

e2tree documentation built on May 15, 2026, 5:06 p.m.