Given a set of input images, find the the class of each input image and the output of the model with respect to each input image. For each input

image, the model should return a vector with predictions. return that prediction along with the index of the entry with the maximum value $($i$.$e$.$

the class number of that input image$)$

HINT $1$: refer to the notebooks given with the slides $($DigitsRecognition$.$ipynb and FashionMNIST.ipynb$)$Given a set of input images, find the the class of each input image and the output of the model with respect to each input image. For each input image, the model should return a vector with predictions. return that prediction along with the index of the entry with the maximum value $($i$.$e$.$ the class number of that input image$)$

HINT $1$: refer to the notebooks given with the slides $($DigitsRecognition$.$ipynb and FashionMNIST.ipynb$)$ # Function to find the index of the largest entry in one prediction vector

def find$\_$class$($model$,$ input$\_$data$)$:

$\text{'}\text{'}\text{'}$

input

model: a trained CNN model used for classification

input$\_$data: a set of input images. Each entry is an input image that needs to be classified

output:

classes: a vector of integers where each entry indicate the class of the corresponding input entry in the input$\_$data

prediction: a vector of $1$D arrays where each entry contains the output of the model $($an array of $10$ entries$)$ corresponding to each entry in the input$\_$data

$\text{'}\text{'}\text{'}$

### BEGIN SOLUTION

return classes, prediction # Returns the index of the largest entry in one prediction vector