Menu

Predicting Image Classes Using ANN

Predicting Image Classes Using ANN

Use the trained ANN model to predict image classes.

Code

predictions = ann.predict(X_test)

Explanation

The model predicts the probability of each image belonging to the ten CIFAR-10 classes. The class with the highest probability is selected as the predicted label.