How are they different? This might indicate where you got it wrong. Maybe when predicting, you set the axis= argument wrong? (the argmax should be on the last axis).
Cheers
How are they different? This might indicate where you got it wrong. Maybe when predicting, you set the axis= argument wrong? (the argmax should be on the last axis).
Cheers