a37910cf0c8fbd1f863c058912571902198cad44,qanta/reporting/plotting.py,,plot_confusion,#Any#Any#Any#Any#,7

Before Change

    plt.colorbar()
    tick_marks = np.arange(len(labels))
    plt.xticks(tick_marks, labels, rotation=45)
    plt.yticks(tick_marks, labels)
    plt.ylabel("True Label")
    plt.xlabel("Predicted Label")
    ax = plt.gca()
    ax.grid(False)

After Change

    else:
        cm = confusion_matrix(true_labels, predicted_labels, labels=labels)

    fig, ax = plt.subplots(figsize=(10, 10))
    ax.imshow(cm, interpolation="nearest", cmap=plt.cm.Blues)
    ax.set_title(title)
    // plt.colorbar()

Instances