mlr3 (version 0.3.0)

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.

Arguments

Super class

mlr3::Prediction -> PredictionRegr

Active bindings

response

(numeric()) Access the stored predicted response.

se

(numeric()) Access the stored standard error.

missing

(integer()) Returns row_ids for which the predictions are missing or incomplete.

Methods

Public methods

Method new()

Creates a new instance of this R6 class.

Usage

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

Arguments

task

(TaskRegr) Task, used to extract defaults for row_ids and truth.

row_ids

(integer()) Row ids of the predicted observations, i.e. the row ids of 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.

See Also

Other Prediction: PredictionClassif, Prediction

Examples

Run this code
# 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))
# }

Run the code above in your browser using DataLab