Compare commits
1 Commits
d0220ab1f0
...
1c5b730cfb
Author | SHA1 | Date |
---|---|---|
coolneng | 1c5b730cfb |
|
@ -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