setPredictType: Set the type of predictions the learner should return.

Description Usage Arguments Value See Also

Description

Possible prediction types are: Classification: Labels or class probabilities (including labels). Regression: Numeric or response or standard errors (including numeric response). Survival: Linear predictor or survival probability.

For complex wrappers the predict type is usually also passed down the encapsulated learner in a recursive fashion.

Usage

1
setPredictType(learner, predict.type)

Arguments

learner

[Learner | character(1)]
The learner. If you pass a string the learner will be created via makeLearner.

predict.type

[character(1)]
Classification: “response” or “prob”. Regression: “response” or “se”. Survival: “response” (linear predictor) or “prob”. Clustering: “response” or “prob”. Default is “response”.

Value

[Learner].

See Also

Other predict: asROCRPrediction, getPredictionProbabilities, getPredictionResponse, plotViperCharts, predict.WrappedModel, setPredictThreshold

Other learner: LearnerProperties, getClassWeightParam, getHyperPars, getLearnerId, getLearnerPackages, getLearnerParVals, getLearnerParamSet, getLearnerPredictType, getLearnerShortName, getLearnerType, getParamSet, makeLearners, makeLearner, removeHyperPars, setHyperPars, setId, setLearnerId, setPredictThreshold



Search within the mlr package
Search all R packages, documentation and source code

Questions? Problems? Suggestions? or email at ian@mutexlabs.com.

Please suggest features or report bugs with the GitHub issue tracker.

All documentation is copyright its authors; we didn't write any of that.