get_multi_class_statistics

get_multi_class_statistics.py

def get_multi_class_stats(statistics_counts, predicted_label_id, actual_label_id, id2label):
    actual_label = id2label[actual_label_id]
    statistics_counts[actual_label]['total'] += 1
    predicted_label = id2label[predicted_label_id]
    if predicted_label == actual_label:
        statistics_counts[actual_label]['true_positives'] += 1
    else:
        # wrong prediction, the prediction is thus a false positive
        # and the actual label is a false negative
        statistics_counts[predicted_label]['false_positives'] += 1
        statistics_counts[actual_label]['false_negatives'] += 1

    return statistics_counts