diff --git a/DashAI/back/models/scikit_learn/svc.py b/DashAI/back/models/scikit_learn/svc.py index d97b7a43c..e52cd0601 100644 --- a/DashAI/back/models/scikit_learn/svc.py +++ b/DashAI/back/models/scikit_learn/svc.py @@ -119,21 +119,6 @@ class SVCSchema(BaseSchema): ), alias=MultilingualString(en="max iterations", es="max iteraciones"), ) # type: ignore - probability: schema_field( - bool_field(), - placeholder=True, - description=MultilingualString( - en=( - "The parameter 'probability' indicates whether or not " - "to predict with probabilities." - ), - es=( - "El parĂ¡metro 'probabilidad' indica si " - "se debe predecir o no con probabilidades." - ), - ), - alias=MultilingualString(en="probability", es="probabilidad"), - ) # type: ignore shrinking: schema_field( bool_field(), placeholder=True, @@ -205,4 +190,5 @@ class SVC(TabularClassificationModel, SklearnLikeClassifier, _SVC): CATEGORICAL_ENCODING = CategoricalEncodingStrategy.ONE_HOT def __init__(self, **kwargs): + kwargs["probability"] = True super().__init__(**kwargs)