Skip to content

Commit

Permalink
Merge pull request #328 from NVIDIA/branch-23.06
Browse files Browse the repository at this point in the history
patch for 23.06 release
  • Loading branch information
YanxuanLiu authored Jul 12, 2023
2 parents ff9817b + b24ed7e commit 04dffdf
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions python/src/spark_rapids_ml/regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -520,7 +520,6 @@ def _single_fit(init_parameters: Dict[str, Any]) -> Dict[str, Any]:
# and label columns are all standardized.
init_parameters = init_parameters.copy()
if "alpha" in init_parameters.keys():
print(f"pdesc.m {pdesc.m}")
init_parameters["alpha"] *= (float)(pdesc.m)

else:
Expand Down Expand Up @@ -708,7 +707,6 @@ def _construct_lr() -> CumlT:

for i in range(len(coefs)):
lr = LinearRegressionMG(output_type="numpy")
print(f"index: {i}: coef: {coefs[i]}")
lr.coef_ = cudf_to_cuml_array(
np.array(coefs[i], order="F").astype(dtype)
)
Expand Down

0 comments on commit 04dffdf

Please sign in to comment.