set_threshold: For a binary classification problem, update the the predicted...

In iai: Interface to 'Interpretable AI' Modules

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)

iai documentation built on July 9, 2023, 5:41 p.m.