From 08c8b063aeda68b7c66395c5879fb18e80377a88 Mon Sep 17 00:00:00 2001
From: "michael.divia" <michael.divia@etu.hesge.ch>
Date: Wed, 9 Apr 2025 18:17:16 +0200
Subject: [PATCH] Migrated from SoftMax to Logits
---
python/pokedex_test.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/pokedex_test.py b/python/pokedex_test.py
index 7cefe57..011ab29 100644
--- a/python/pokedex_test.py
+++ b/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]
--
GitLab