Compare commits
1 Commits
1c5b730cfb
...
d0220ab1f0
Author | SHA1 | Date |
---|---|---|
coolneng | d0220ab1f0 |
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue