Skip to content

ModelConfigPM

Bases: ExtraBaseModel

Pydantic model for SimpleModel config.

Attributes:

Name Type Description
models_dir str

Directory where models are stored. Defaults to "models".
modelName (str

Name of the model which is also the sub-directory name. Defaults to "linear_regression.v0.0.1-240101".
threshold (float

Threshold value for the similarity score. Defaults to None.
Source code in src/simple_model/config.py 
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class ModelConfigPM(ExtraBaseModel):
    """Pydantic model for SimpleModel config.

    Attributes:
        models_dir (str  ): Directory where models are stored. Defaults to "models".
        modelName  (str  ): Name of the model which is also the sub-directory name. Defaults to "linear_regression.v0.0.1-240101".
        threshold  (float): Threshold value for the similarity score. Defaults to None.
    """

    models_dir: constr(strip_whitespace=True) = Field(  # type: ignore
        default="./models", min_length=2, max_length=1023
    )
    modelName: constr(strip_whitespace=True) = Field(  # type: ignore
        default="linear_regression.v0.0.1-240101",
        min_length=2,
        max_length=128,
        alias="model_name",
    )
    threshold: float = Field(default=0.5, ge=0.0, le=1.0)