PredictionRegr: Prediction Object for Regression

In mllg/mlr3: Machine Learning in R - Next Generation

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

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))

mllg/mlr3 documentation built on Sept. 27, 2019, 9:38 a.m.