iai (version 1.0.0)

set_threshold: set_threshold

Description

For a binary classification problem, update the the predicted labels in the leaves of the learner to predict a label only if the predicted probability is at least the specified threshold.

Usage

set_threshold(lnr, label, threshold, ...)

Arguments

lnr

The learner or grid to modify.

label

The referenced label.

threshold

The probability threshold above which label will be be predicted.

...

Refer to the Julia documentation for available parameters.

Details

Julia Equivalent: IAI.set_threshold!

Examples

# NOT RUN {
iai::set_threshold(lnr, "A", 0.4)
# }
# NOT RUN {
# }