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