coolneng/locimend
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add AUC metric to model

Browse Source
This commit is contained in:
coolneng
2021-07-06 03:45:17 +02:00
parent c24f528484
commit d0220ab1f0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/model.py
View File

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