Fix typo in model_v2.py

2022-12-07 00:08:19 +01:00 · 2022-12-07 00:08:19 +01:00 · 6d3c2290b2
parent 94ca84b4cd
commit 6d3c2290b2
1 changed files with 1 additions and 1 deletions
--- a/model_v2.py
+++ b/model_v2.py
@ -270,7 +270,7 @@ def train_model(
        np.average(hypermodel_history.history["val_loss"]),
    )

-    return model
+    return hypermodel


 def test_model(model):