candle.uq_keras_utils.add_index_to_output

candle.uq_keras_utils.add_index_to_output(y_train)

This function adds a column to the training output to store the indices of the corresponding samples in the training set.

Parameters:

y_train (ndarray) – Numpy array of the output in the training set