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

SIENTIAPDE-1255: Integrate sientia-mlops-library into model-manager, adding model serving, reporting, and updated model definitions.

Browse Source
This commit is contained in:
Bruno Domingues
2025-10-20 16:55:52 -03:00
parent 9e31ab679b
commit 56a21a16da
12 changed files with 998 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
model_manager/utils/models/train_model_result.py
View File

@@ -1,9 +1,8 @@
from dataclasses import dataclass
import pandas as pd
from sientia.linear_models import LinearRegressionModel
from sientia.preprocessing import DataPreprocessor
from model_manager.sientia.models import DataPreprocessor, LinearRegressionModel
from model_manager.utils.models.train_model_params import TrainModelParams
@@ -40,8 +39,8 @@ class TrainModelResult:
process_data: DataPreprocessor
x_train: pd.DataFrame
x_test: pd.DataFrame
y_train: pd.DataFrame
y_test: pd.DataFrame
y_train: pd.Series
y_test: pd.Series
regr: LinearRegressionModel
scaler_dict: dict
y_pred: pd.Series | None = None