Skip to content

Commit cee8948

Browse files
committed
feat: enhance multilingual support in LogisticRegression and its schema
1 parent 5572f9c commit cee8948

1 file changed

Lines changed: 36 additions & 7 deletions

File tree

DashAI/back/models/scikit_learn/logistic_regression.py

Lines changed: 36 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
optimizer_int_field,
88
schema_field,
99
)
10+
from DashAI.back.core.utils import MultilingualString
1011
from DashAI.back.models.scikit_learn.sklearn_like_classifier import (
1112
SklearnLikeClassifier,
1213
)
@@ -25,7 +26,11 @@ class LogisticRegressionSchema(BaseSchema):
2526
penalty: schema_field(
2627
enum_field(enum=["l2", "l1", "elasticnet"]),
2728
placeholder="l2",
28-
description="Specify the norm of the penalty",
29+
description=MultilingualString(
30+
en="Specify the norm of the penalty",
31+
es="Especifica la norma de la penalización",
32+
),
33+
alias=MultilingualString(en="Penalty", es="Penalización"),
2934
) # type: ignore
3035
tol: schema_field(
3136
optimizer_float_field(ge=0.0),
@@ -35,7 +40,11 @@ class LogisticRegressionSchema(BaseSchema):
3540
"lower_bound": 0.0,
3641
"upper_bound": 5.0,
3742
},
38-
description="Tolerance for stopping criteria.",
43+
description=MultilingualString(
44+
en="Tolerance for stopping criteria.",
45+
es="Tolerancia para el criterio de detención.",
46+
),
47+
alias=MultilingualString(en="Tolerance", es="Tolerancia"),
3948
) # type: ignore
4049
C: schema_field(
4150
optimizer_float_field(gt=0.0),
@@ -45,8 +54,18 @@ class LogisticRegressionSchema(BaseSchema):
4554
"lower_bound": 1.0,
4655
"upper_bound": 7.0,
4756
},
48-
description="Inverse of regularization strength, smaller values specify "
49-
"stronger regularization. Must be a positive number.",
57+
description=MultilingualString(
58+
en=(
59+
"Inverse of regularization strength, smaller values specify stronger "
60+
"regularization. Must be a positive number."
61+
),
62+
es=(
63+
"Inverso de la fuerza de regularización, valores más pequeños "
64+
"especifican una regularización más fuerte. Debe ser un número "
65+
"positivo."
66+
),
67+
),
68+
alias=MultilingualString(en="C", es="C"),
5069
) # type: ignore
5170
max_iter: schema_field(
5271
optimizer_int_field(ge=50),
@@ -56,7 +75,11 @@ class LogisticRegressionSchema(BaseSchema):
5675
"lower_bound": 50,
5776
"upper_bound": 250,
5877
},
59-
description="Maximum number of iterations taken for the solvers to converge.",
78+
description=MultilingualString(
79+
en=("Maximum number of iterations taken for the solvers to converge."),
80+
es=("Número máximo de iteraciones para que los solucionadores converjan."),
81+
),
82+
alias=MultilingualString(en="Max iterations", es="Máximas iteraciones"),
6083
) # type: ignore
6184

6285

@@ -66,8 +89,14 @@ class LogisticRegression(
6689
"""Scikit-learn's Logistic Regression wrapper for DashAI."""
6790

6891
SCHEMA = LogisticRegressionSchema
69-
DISPLAY_NAME: str = "Logistic Regression"
70-
DESCRIPTION: str = "Linear model for classification using logistic function."
92+
DISPLAY_NAME: str = MultilingualString(
93+
en="Logistic Regression",
94+
es="Regresión Logística",
95+
)
96+
DESCRIPTION: str = MultilingualString(
97+
en="Linear model for classification using logistic function.",
98+
es="Modelo lineal para clasificación usando la función logística.",
99+
)
71100
COLOR: str = "#64B5F6"
72101
ICON: str = "TrendingUp"
73102

0 commit comments

Comments
 (0)