final cleanup

autonomio · Apr 14, 2022 · fed0298 · fed0298
1 parent 914cb0a
commit fed0298
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/commands/test_latest.py b/tests/commands/test_latest.py
@@ -62,6 +62,10 @@ def iris_model(x_train, y_train, x_val, y_val, params):
                              reduction_method='gamify',
                              save_models=True)
 
+    scan_object.best_model(saved=True)
+    predict = talos.Predict(scan_object)
+    predict.predict(x, 'val_acc', False, saved=True)
+
     scan_object = talos.Scan(x, y,
                              model=iris_model,
                              params=p,