Migrated from SoftMax to Logits

python/pokedex_test.py

@@ -54,7 +54,7 @@ for i in range(4):
 
     # --- Predict ---
     predictions = model.predict(img_array, verbose=0)
-    probabilities = predictions[0]
+    probabilities = tf.nn.softmax(predictions[0])
     predicted_class_index = np.argmax(probabilities)
     predicted_label = class_names[predicted_class_index]
     confidence = 100 * probabilities[predicted_class_index]