I've created a CNN in RStudio using keras to predict MNIST digits. I am now trying to predict with this model using the following code
cnn_pred <- cnn_model %>%
predict_classes(x_test)
but predict_classes() has been deprecated and I need something to replace it. I've tried using just predict() but it results in continuous predictions when I need it to predict what the digit is (0, 1, 2, 3, 4, 5, 6, 7, 8, or 9).
What function can be used with the CNN model to give a categorical prediction?