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

SIENTIAPDE-1171

Browse Source 
Refactor data merging in MLFlowRepository to simplify merge operation in fit method. Removed unnecessary parameters for improved readability and maintainability.
This commit is contained in:
vitor-aignosi
2025-07-24 11:24:21 -03:00
parent 1261e4a8aa
commit a34bdcd839
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
laborious/utils/repository/model_repository.py
View File

@@ -135,13 +135,7 @@ class MLFlowRepository():
target_name = data_model.target_variable target_name = data_model.target_variable
y = data[target_name] y = data[target_name]
treated_data = pd.merge( treated_data = pd.merge(
treated_data, y, treated_data, y, left_index=True, right_index=True)
left_index=True,
right_index=True,
on='id',
how='left', # or 'inner', depending on your needs
validate='one_to_one' # ensures each index appears only once in both
)
prediction_model = prediction_model.fit(treated_data) prediction_model = prediction_model.fit(treated_data)
experiment = self.get_experiment_by_run_id(latest_production_id) experiment = self.get_experiment_by_run_id(latest_production_id)
mlflow.set_experiment(experiment) mlflow.set_experiment(experiment)