From 102a58025b08bef1632440389a892dfd0b0d8222 Mon Sep 17 00:00:00 2001 From: Erik Ordentlich Date: Thu, 8 Jan 2026 07:50:20 -0800 Subject: [PATCH] avoid collisions in logistic_regression_objective tmp column name Signed-off-by: Erik Ordentlich --- python/src/spark_rapids_ml/metrics/utils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/python/src/spark_rapids_ml/metrics/utils.py b/python/src/spark_rapids_ml/metrics/utils.py index 6b6d99ea..98c6c3c6 100644 --- a/python/src/spark_rapids_ml/metrics/utils.py +++ b/python/src/spark_rapids_ml/metrics/utils.py @@ -54,9 +54,14 @@ def logistic_regression_objective( # account for effects of standardization on the coefficients for regularization penalty if lr_model.getStandardization() is True: + column_names = df.columns + outputCol = "_objective_tmp" + while outputCol in column_names: + outputCol = "_" + outputCol + scaler = StandardScaler( inputCol=features_col, - outputCol="scaledFeatures", + outputCol=outputCol, ) scaler_model = scaler.fit(df) stdev = np.array(scaler_model.std)