Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 300a1a81 authored by ANDREY Paul's avatar ANDREY Paul Committed by BIGAUD Nathan
Browse files

Document 'penalty="none"' option for 'SklearnSGDModel'.

parent 478d8c4e
No related branches found
No related tags found
1 merge request!28Functional test of regression
Pipeline #764613 passed
......@@ -151,7 +151,7 @@ class SklearnSGDModel(Model):
cls,
kind: Literal["classifier", "regressor"],
loss: Optional[LossesLiteral] = None,
penalty: Literal["l1", "l2", "elasticnet"] = "l2",
penalty: Literal["none", "l1", "l2", "elasticnet"] = "l2",
alpha: float = 1e-4,
l1_ratio: float = 0.15,
epsilon: float = 0.1,
......@@ -173,7 +173,7 @@ class SklearnSGDModel(Model):
See `sklearn.linear_model.SGDRegressor` and `SGDClassifier`
documentation for details on possible values. If None, set
to "hinge" for classifier or "squared_error" for regressor.
penalty: {"l1", "l2", "elasticnet"}, default="l2"
penalty: {"none", "l1", "l2", "elasticnet"}, default="l2"
The penalty (i.e. regularization term) to be used.
alpha: float, default=0.0001
Regularization constant (the higher the stronger).
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment