Compare commits

..

1 Commits

Author SHA1 Message Date
coolneng 1c5b730cfb
Add AUC metric to model 2021-07-06 03:45:17 +02:00
1 changed files with 1 additions and 1 deletions

View File

@ -40,7 +40,7 @@ def build_model(hyperparams) -> Model:
model.compile(
optimizer=Adam(hyperparams.learning_rate),
loss=categorical_crossentropy,
metrics=["accuracy", "AUC"],
metrics=["accuracy" "AUC"],
)
return model