def print_confusion_matrix()

in mapillary_vistas/helpers/print.py [0:0]

• 24 lines of code
• 6 McCabe index (conditional complexity)

def print_confusion_matrix(labels, confusion_matrix, percent=False):
    max_length = get_max_length(labels)

    print('')
    print('')
    print('')

    header = "{:>{ml}} |".format("", ml=max_length+1)
    for label in labels:
        header += "{:>{ml}} |".format(label["name"], ml=max_length)
    print(header)

    for index, label in enumerate(labels):
        current_line = "{:>{ml}}: |".format(label["name"], ml=max_length)
        divider = float(np.sum(confusion_matrix[index, :]))
        for inner_index, _ in enumerate(labels):
            if percent:
                if divider == 0:
                    metric = float('nan')
                else:
                    metric = confusion_matrix[index, inner_index] / divider
                value = "{:.2%}".format(metric)
            else:
                value = "{:d}".format(confusion_matrix[index, inner_index])
            current_line += "{:>{ml}} |".format(value, ml=max_length)
        print(current_line)

    print()