Skip to content

Commit 2e3aefe

Browse files
Fix custom model recipe
1 parent efce143 commit 2e3aefe

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

bluecast/blueprints/custom_model_recipes.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,14 +38,17 @@ def autotune(
3838
"penalty": ["l2"],
3939
"C": np.logspace(0.1, 1, 5),
4040
"class_weight": ["balanced", None],
41-
"solver": ["newton-cg", "newton-cholesky", "sag", "saga"],
41+
# solvers that support l2
42+
"solver": ["lbfgs", "newton-cg", "newton-cholesky", "sag", "saga"],
4243
},
4344
{
4445
"penalty": ["elasticnet"],
4546
"C": np.logspace(0.1, 1, 5),
4647
"class_weight": ["balanced", None],
47-
"solver": ["newton-cg", "newton-cholesky", "sag", "saga"],
48-
"l1_ratio": np.arange(0, 1, 3),
48+
# elasticnet is only supported by 'saga'
49+
"solver": ["saga"],
50+
# include endpoints 0.0, 0.5, 1.0
51+
"l1_ratio": np.linspace(0.0, 1.0, 3),
4952
},
5053
]
5154

0 commit comments

Comments
 (0)