SIENTIAPDE-1171

Enhance data merging in MLFlowRepository by specifying the 'on' parameter for improved join accuracy. This change ensures that the merge operation correctly aligns data based on the 'id' field.
This commit is contained in:
vitor-aignosi
2025-07-24 10:56:05 -03:00
parent a2492e244d
commit 1261e4a8aa

View File

@@ -138,6 +138,7 @@ class MLFlowRepository():
treated_data, y, treated_data, y,
left_index=True, left_index=True,
right_index=True, right_index=True,
on='id',
how='left', # or 'inner', depending on your needs how='left', # or 'inner', depending on your needs
validate='one_to_one' # ensures each index appears only once in both validate='one_to_one' # ensures each index appears only once in both
) )