Activity 5 - Testing the Model

Workshop Resources

Now that we have built and trained our model, we want to go ahead and test it against returned clothing items. Let’s start off with our first sample item, which was the shoe.

probability_model = tf.keras.Sequential([model, tf.keras.layers.Softmax()])
predictions = probability_model.predict(test_images)

The following code gives a list of probabilities that an image fits in a specific category. The index value corresponds to what image would be classified.

predictions[0] #Returns the index with highest probability, depends on the index value.

Question 1

The array of values represents the confidence level that an image belongs to a specific category. What index has the most confidence value, and what has the least confidence value?

Question 2

Verify your answer to the highest probability value by running the code below:

np.argmax(predictions[0])
test_labels[0]

In the class_names array, what category does this index correspond to? (Hint: Given the index of the maximum probability value, what index should you look for?)

Verify what category this image falls under using this code. Is it correct?

class_names[9]