From d0220ab1f043c195da1797d29adf3b16e8311a4c Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 6 Jul 2021 03:45:17 +0200
Subject: [PATCH] Add AUC metric to model

---
 src/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/model.py b/src/model.py
index e163c6e..1b451ab 100644
--- a/src/model.py
+++ b/src/model.py
@@ -40,7 +40,7 @@ def build_model(hyperparams) -> Model:
     model.compile(
         optimizer=Adam(hyperparams.learning_rate),
         loss=categorical_crossentropy,
-        metrics=["accuracy"],
+        metrics=["accuracy", "AUC"],
     )
     return model