Menu

Predicting Image Classes

Predicting Image Classes

Generate predictions using the trained CNN.

Code

predictions = cnn.predict(X_test)

Explanation

The model predicts the probability for each class. The class with the highest probability is selected as the final prediction.