diff --git a/src/jenga/basis.py b/src/jenga/basis.py index 806ad8d..d56912c 100644 --- a/src/jenga/basis.py +++ b/src/jenga/basis.py @@ -300,7 +300,7 @@ def _get_pipeline_grid_scorer_tuple( """ param_grid = { - 'learner__loss': ['log'], + 'learner__loss': ['log_loss'], 'learner__penalty': ['l2'], 'learner__alpha': [0.00001, 0.0001, 0.001, 0.01] } @@ -419,7 +419,7 @@ def _get_pipeline_grid_scorer_tuple( """ param_grid = { - 'learner__loss': ['log'], + 'learner__loss': ['log_loss'], 'learner__penalty': ['l2'], 'learner__alpha': [0.00001, 0.0001, 0.001, 0.01] }