candle.uq_keras_utils.abstention_acc_class_i_metric

candle.uq_keras_utils.abstention_acc_class_i_metric(nb_classes, class_i)

Function to estimate accuracy over the class i prediction after removing the samples where the model is abstaining.

Parameters:

• nb_classes (int or ndarray) – Integer or numpy array defining indices of the abstention class

• class_i (int) – Index of the class to estimate accuracy after removing abstention samples