mlr3 (version 0.1.4)

PredictionRegr: Prediction Object for Regression

Description

This object wraps the predictions returned by a learner of class LearnerRegr, i.e. the predicted response and standard error.

Format

R6::R6Class object inheriting from Prediction.

Construction

p = PredictionRegr$new(task = NULL, row_ids = task$row_ids, truth = task$truth(), response = NULL, se = NULL)

• task :: TaskRegr Task, used to extract defaults for row_ids and truth.

• row_ids :: (integer() | character()) Row ids of the observations in the test set.

• truth :: numeric() True (observed) response.

• response :: numeric() Vector of numeric response values. One element for each observation in the test set.

• se :: numeric() Numeric vector of predicted standard errors. One element for each observation in the test set.

Fields

All fields from Prediction, and additionally:

• response :: numeric() Access to the stored predicted response.

• se :: numeric() Access to the stored standard error.

The field task_type is set to "regr".

See Also

Other Prediction: PredictionClassif, Prediction

Examples

# NOT RUN {
task = tsk("boston_housing")
learner = lrn("regr.featureless", predict_type = "se")
p = learner$train(task)$predict(task)
p$predict_types
head(as.data.table(p))
# }