# python - keras model only predicts one class for all the

1.Changed the class_mode to 'categorical' for the train and test generators 2.Changed the final dense layer from 1 to 2 so this will return scores/probabilities for both classes. So when you use argmax, it will return the index position of the top score indicating which class it has predicted. Share