model compile

gistfile1.txt

# The from_logits=True attribute inform the loss function that the output values generated by the model are not normalized, a.k.a. logits.
# Since softmax layer is not being added at the last layer, hence we need to have the from_logits=True to indicate the probabilities are not normalized.

model.compile(optimizer= 'adam', 
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
              metrics = ['accuracy'])