diff --git a/network-operations/src/workflow.py b/network-operations/src/workflow.py index 395899e2..43b1e821 100644 --- a/network-operations/src/workflow.py +++ b/network-operations/src/workflow.py @@ -48,7 +48,7 @@ def pipeline( train = mlrun.run_function( auto_trainer, handler='train', - params={"model_class": model_pkg_class}, + params={"model_class": model_pkg_class, "random_state": 3}, inputs={"dataset": get_vector.outputs['feature_vector']}, outputs=["model", "test_set"], )