2118f87a1d95e673ac0994989a5464480ce2ffe1,pycm/pycm_obj.py,ConfusionMatrix,plot,#ConfusionMatrix#,884

Before Change

        matrix = self.to_array(normalized=normalized)
        if normalized:
            title += " (Normalized)"
        classes = [self.label_map[_class] for _class in sorted(self.label_map.keys())]
        if plot_lib == "matplotlib":
            import matplotlib.pyplot as plt
            fig, ax = plt.subplots()

After Change

        if normalized:
            title += " (Normalized)"
        classes = sorted(self.classes)
        if one_vs_all and class_name in classes:
            classes = [class_name, "~"]
        if plot_lib == "matplotlib":
            from matplotlib import pyplot as plt
            fig, ax = plt.subplots()
            plt.imshow(matrix, cmap=plt.cm.gray_r)

Instances