7a7d0e7b35cbee6b3409ac30ff4bf4edfe525af2,alphapy/plots.py,,plot_confusion_matrix,#Any#Any#,593

Before Change

        cmn = cm.astype("float") / cm.sum(axis=1)[:, np.newaxis]
        // place text in square of confusion matrix
        thresh = (cm.max() + cm.min()) / 2.0
        for i, j in product(list(range(cm.shape[0])), list(range(cm.shape[1]))):
            cmr = round(cmn[i, j], 3)
            plt.text(j, i, cmr,
                     horizontalalignment="center",
                     color="white" if cm[i, j] > thresh else "black")
        // labels
        plt.tight_layout()
        plt.ylabel("True Label")
        plt.xlabel("Predicted Label")
        // save the chart

After Change

                        color="white" if cm[i, j] > thresh else "black")

        // show the color bar
        im = ax.imshow(cm, interpolation="nearest", cmap=cmap)
        ax.figure.colorbar(im, ax=ax)

        // save the chart
        tag = USEP.join([pstring, algo])
        write_plot("matplotlib", plt, "confusion", tag, plot_dir)

Instances