set_threshold {iai}

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.

Description

Julia Equivalent: IAI.set_threshold!

Usage

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

Arguments

lnr

The learner 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.

Examples

## Not run: iai::set_threshold(lnr, "A", 0.4)