setPredictThreshold: Set the probability threshold the learner should use.
Description
See predict.threshold in makeLearner and setThreshold. For complex wrappers only the top-level predict.type is currently set.Usage
setPredictThreshold(learner, predict.threshold)
Arguments
learner
[Learner | character(1)]
The learner.
If you pass a string the learner will be created via makeLearner. predict.threshold
[numeric]
Threshold to produce class labels. Has to be a named vector, where names correspond to class labels.
Only for binary classification it can be a single numerical threshold for the positive class.
See setThreshold for details on how it is applied.
Default is NULL which means 0.5 / an equal threshold for each class. See Also
Other predict: asROCRPrediction,
getPredictionProbabilities,
getPredictionResponse,
plotViperCharts,
predict.WrappedModel,
setPredictType Other learner: LearnerProperties,
getClassWeightParam,
getHyperPars, getLearnerId,
getLearnerPackages,
getLearnerParVals,
getLearnerParamSet,
getLearnerPredictType,
getLearnerShortName,
getLearnerType, getParamSet,
makeLearners, makeLearner,
removeHyperPars,
setHyperPars, setId,
setLearnerId, setPredictType