Learn R Programming

iml (version 0.1)

prediction.model: Get a prediction object

Description

Get a prediction object

Usage

prediction.model(object, class = NULL, predict.args = NULL)

Arguments

object

function, mlr WrappedObject, S3 class with predict function, or caret train object

class

In case of classification, class specifies the class for which to predict the probability. By default the first class in the prediction (first column) is chosen.

predict.args

named list with arguments passed down to the prediction model

Value

object of type Prediction