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