Optimize hyperparameters, Selecting the best model and Testing it

ASabryMazroua · Jun 30, 2021 · bd9a5eb · bd9a5eb
1 parent 351c951
commit bd9a5eb
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/main.py b/main.py
@@ -111,12 +111,14 @@ def go(config: DictConfig):
             )
 
         if "test_regression_model" in active_steps:
-
-            ##################
-            # Implement here #
-            ##################
-
-            pass
+            _ = mlflow.run(
+                f"{config['main']['components_repository']}/test_regression_model",
+                "main",
+                parameters={
+                    "mlflow_model": "random_forest_export:prod",
+                    "test_dataset": "test_data.csv:latest",
+                },
+            )
 
 
 if __name__ == "__main__":