Add AUC metric to model

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