From 1c5b730cfb1808ef634244c256083ca912e50e5f 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..1af46ba 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