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

Add AUC metric to model

This commit is contained in:
coolneng 2021-07-06 03:45:17 +02:00
parent c24f528484
commit d0220ab1f0
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 1 additions and 1 deletions
Show Stats 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