How to predict a class after training the model?
How to predict a class after training the model?