[madlib] 01/06: Elastic-net: Disable ORCA to improve exec time

[ok...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

okislal pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/madlib.git

commit 5174627afa9122c7d6aa17479fcd7a0e4dfd4cf4
Author: Orhan Kislal <ok...@apache.org>
AuthorDate: Wed Apr 27 14:15:22 2022 +0300

    Elastic-net: Disable ORCA to improve exec time
---
 src/ports/postgres/modules/elastic_net/elastic_net.py_in | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ports/postgres/modules/elastic_net/elastic_net.py_in b/src/ports/postgres/modules/elastic_net/elastic_net.py_in
index c6dec53e..4364aa89 100644
--- a/src/ports/postgres/modules/elastic_net/elastic_net.py_in
+++ b/src/ports/postgres/modules/elastic_net/elastic_net.py_in
@@ -8,6 +8,7 @@ from elastic_net_utils import _generate_warmup_lambda_sequence
 
 from elastic_net_utils import BINOMIAL_FAMILIES, GAUSSIAN_FAMILIES, OPTIMIZERS
 
+from utilities.control import OptimizerControl
 from utilities.validate_args import is_col_array
 from utilities.utilities import is_string_formatted_as_array_expression
 from utilities.validate_args import table_exists
@@ -342,7 +343,8 @@ def elastic_net_train(schema_madlib, source_table, model_table, dependent_varnam
         if cv_param['n_folds'] > 1:
             args.update(cv_param)
             _cross_validate_en(args)
-        _internal_elastic_net_train(**args)
+        with OptimizerControl(False):
+            _internal_elastic_net_train(**args)
 # ------------------------------------------------------------------------