We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent efce143 commit 2e3aefeCopy full SHA for 2e3aefe
bluecast/blueprints/custom_model_recipes.py
@@ -38,14 +38,17 @@ def autotune(
38
"penalty": ["l2"],
39
"C": np.logspace(0.1, 1, 5),
40
"class_weight": ["balanced", None],
41
- "solver": ["newton-cg", "newton-cholesky", "sag", "saga"],
+ # solvers that support l2
42
+ "solver": ["lbfgs", "newton-cg", "newton-cholesky", "sag", "saga"],
43
},
44
{
45
"penalty": ["elasticnet"],
46
47
48
- "l1_ratio": np.arange(0, 1, 3),
+ # 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),
52
53
]
54
0 commit comments