From 01e6d09f7cc1cb045bb87d4ae55b927d2367f37d Mon Sep 17 00:00:00 2001
From: 87owo <85057800+87owo@users.noreply.github.com>
Date: Wed, 26 Nov 2025 17:56:08 +0800
Subject: [PATCH] Update train.py

---
 Engine/Model/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Engine/Model/train.py b/Engine/Model/train.py
index 2e7f1cd..76c9c0c 100644
--- a/Engine/Model/train.py
+++ b/Engine/Model/train.py
@@ -127,7 +127,7 @@ with strategy.scope():
         print("Creating a new model")
         model = build_model(train_ds.image_shape, len(train_ds.class_indices))
     print(f"Total parameters: {model.count_params()}")
-    model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
+    model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy', tf.keras.metrics.Recall()])
+
 model.fit(train_ds, epochs=30, callbacks=[CustomModelCheckpoint(), callbacks.LearningRateScheduler(lr_scheduler)], validation_data=val_ds)
 input('Training Complete')
-