I tried to visualize the actual images with the predictions to figure out how my algorithm is performing and which labels are being predicted incorrectly. But when I set the