ec151176c159d79f733ab1255f299dfddfcf801b,tf_unet/util.py,,plot_prediction,#Any#Any#Any#Any#,15

Before Change

            ax[i, 0].set_title("x")
            ax[i, 1].set_title("y")
            ax[i, 2].set_title("pred 0")
            ax[i, ].set_title("pred 1")
    fig.tight_layout()
    
    if save:

After Change

        plt.colorbar(cax, ax=ax[i,1])
        pred = prediction[i, ..., 1]
        pred -= np.amin(pred)
        pred /= np.amax(pred)
        cax = ax[i, 2].imshow(pred)
        plt.colorbar(cax, ax=ax[i,2])
        if i==0:
            ax[i, 0].set_title("x")

Instances