classification_report.utils

Module Contents

classification_report.utils.convert_prob_to_label(data: np.ndarray) → np.ndarray

Convert Probability to labels

Parameters:data – (np.ndarray): Probability output of softmax. The size of data (batch_size, num_classes) Returns:Returns pobability converted to labels by finding maximum in last dimension. The output shape is (batch_size,) Return type:np.ndarray