Convert object to a PredictionRegr.

as_prediction_regr(x, ...)

# S3 method for PredictionRegr
as_prediction_regr(x, ...)

# S3 method for data.frame
as_prediction_regr(x, ...)

Arguments

x

(any)
Object to convert.

...

(any)
Additional arguments.

Value

PredictionRegr.

Examples

# create a prediction object task = tsk("mtcars") learner = lrn("regr.rpart") learner$train(task) p = learner$predict(task) # convert to a data.table tab = as.data.table(p) # convert back to a Prediction as_prediction_regr(tab)
#> <PredictionRegr> for 32 observations: #> row_ids truth response #> 1 21.0 18.26429 #> 2 21.0 18.26429 #> 3 22.8 26.66364 #> --- #> 30 19.7 18.26429 #> 31 15.0 13.41429 #> 32 21.4 26.66364
# split data.table into a list of data.tables tabs = split(tab, cut(tab$truth, 3)) # convert back to list of predictions preds = lapply(tabs, as_prediction_regr) # calculate performance in each group sapply(preds, function(p) p$score())
#> (10.4,18.2].regr.mse (18.2,26.1].regr.mse (26.1,33.9].regr.mse #> 4.278393 9.122466 22.719322