Skip to content

Commit 45d0ae7

Browse files
committed
typos and update comment in bench script
Signed-off-by: Erik Ordentlich <[email protected]>
1 parent 83c0148 commit 45d0ae7

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/benchmark/benchmark/bench_linear_regression.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,8 @@ def run_once(
114114
)
115115

116116
# note: results for spark ML and spark rapids ml will currently match in all regularization
117-
# cases only if features and labels were standardized in the original dataset. Otherwise,
118-
# they will match only if regParam = 0 or elastNeticParam = 1.0 (aka Lasso)
117+
# cases only if features and labels were standardized in the original dataset or if standardization is enabled.
118+
# Otherwise, they will match only if regParam = 0 or elasticNetParam = 1.0 (aka Lasso)
119119
print(
120120
f"RMSE: {rmse}, coefs l1: {coefs_l1}, coefs l2^2: {coefs_l2}, "
121121
f"full_objective: {full_objective}, intercept: {model.intercept}"

python/src/spark_rapids_ml/regression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -311,7 +311,7 @@ class LinearRegression(
311311
-----
312312
Results for spark ML and spark rapids ml fit() will currently be close in all regularization
313313
cases only if features and labels are standardized in the input dataframe or when standardization is enabled. Otherwise,
314-
they will be close only if regParam = 0 or elastNeticParam = 1.0 (aka Lasso).
314+
they will be close only if regParam = 0 or elasticNetParam = 1.0 (aka Lasso).
315315
316316
Parameters
317317
----------

0 commit comments

Comments
 (0)