What are the two images showing?
Is the first one plotting the labels and the second one the predictions?
If so, it seems the plot function or a reshaping is wrong, since the target labels are already visualized out of bounds.