Can't build the output

[The-Engineer9]

Hi,

I have trained the model and I am trying to visualize the predictions, however I am having trouble. The error occurs on the second last line; output = apply_color_map(np.argmax(y[0], axis=-1), labels) I also noticed that there is y[1] and y[3], as well.

Can anyone help me please?