candle.uq_keras_utils.modify_labels

candle.uq_keras_utils.modify_labels(numclasses_out, ytrain, ytest, yval=None)

This function generates a categorical representation with a class added for indicating abstention.

Parameters:

• numclasses_out (int) – Original number of classes + 1 abstention class

• ytrain (ndarray) – Numpy array of the classes (labels) in the training set

• ytest (ndarray) – Numpy array of the classes (labels) in the testing set

• yval (ndarray) – Numpy array of the classes (labels) in the validation set