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