Atividade 5 - Testando o Modelo
Testando Nosso Modelo
Agora que construímos e treinamos nosso modelo, queremos testá-lo com itens de roupas devolvidos. Vamos começar com o primeiro item de amostra, que era o sapato:
# Inicia um objeto de modelo com camada softmax
probability_model = tf.keras.Sequential([model, tf.keras.layers.Softmax()])
Uma camada softmax é um tipo de camada de rede neural que transforma os dados de entrada em uma distribuição de probabilidade entre 0 e 1.
predictions = probability_model.predict(test_images)
O código abaixo retorna uma lista de probabilidades de que uma imagem pertence a uma categoria específica. O índice corresponde à imagem sendo classificada.
predictions[0] # Retorna o vetor de probabilidades para a primeira imagem de teste
Pergunta 1
O array retornado representa o nível de confiança de que a imagem pertence a cada categoria.
- Qual índice possui o valor de maior confiança?
- E qual o menor valor de confiança?
Pergunta 2
Verifique sua resposta com o seguinte código:
np.argmax(predictions[0])
test_labels[0]
Na lista class_names, a que categoria esse índice corresponde?
(Dica: dado o índice do maior valor de probabilidade, qual índice você deve consultar?)
Verifique a que categoria a imagem pertence com o código:
class_names[9]
Está correto?