gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1321: Formatted files

Browse Source
This commit is contained in:
Kou-Kinoshita
2025-10-29 10:24:25 -03:00
parent 734221e32d
commit 25ad4729f7
3 changed files with 21 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -145,6 +145,7 @@ celerybeat.pid
env/ env/
venv/ venv/
ENV/ ENV/
venv_311/
env.bak/ env.bak/
venv.bak/ venv.bak/

16
model_manager/utils/repository/training_repository.py
View File

@@ -254,7 +254,9 @@ class TrainingRepository:
ar_var=params.target_variable if params.include_ar else None, ar_var=params.target_variable if params.include_ar else None,
) )
def _extract_model_equation(self, regr: LinearRegressionModel, params: TrainModelParams) -> dict: def _extract_model_equation(
self, regr: LinearRegressionModel, params: TrainModelParams
) -> dict:
""" """
Extract the linear regression equation coefficients and create equation metadata. Extract the linear regression equation coefficients and create equation metadata.
@@ -283,12 +285,14 @@ class TrainingRepository:
coefficients_dict[var] = float(coefficients[i]) coefficients_dict[var] = float(coefficients[i])
# Create equation string # Create equation string
equation_parts = [f"{coef:.6f} * {var}" for var, coef in coefficients_dict.items()] equation_parts = [f'{coef:.6f} * {var}' for var, coef in coefficients_dict.items()]
equation_string = f"{params.target_variable} = {intercept:.6f} + " + " + ".join(equation_parts) equation_string = f'{params.target_variable} = {intercept:.6f} + ' + ' + '.join(
equation_parts
)
# Create LaTeX equation # Create LaTeX equation
latex_parts = [f"{coef:.6f} \\cdot {var}" for var, coef in coefficients_dict.items()] latex_parts = [f'{coef:.6f} \\cdot {var}' for var, coef in coefficients_dict.items()]
latex_equation = f"{params.target_variable} = {intercept:.6f} + " + " + ".join(latex_parts) latex_equation = f'{params.target_variable} = {intercept:.6f} + ' + ' + '.join(latex_parts)
return { return {
'target_variable': params.target_variable, 'target_variable': params.target_variable,
@@ -296,5 +300,5 @@ class TrainingRepository:
'intercept': float(intercept), 'intercept': float(intercept),
'equation_string': equation_string, 'equation_string': equation_string,
'latex_equation': latex_equation, 'latex_equation': latex_equation,
'model_type': 'Linear Regression' 'model_type': 'Linear Regression',
} }