-
- Downloads
[SPARK-16500][ML][MLLIB][OPTIMIZER] add LBFGS convergence warning for all used place in MLLib
## What changes were proposed in this pull request? Add warning_for the following case when LBFGS training not actually convergence: 1) LogisticRegression 2) AFTSurvivalRegression 3) LBFGS algorithm wrapper in mllib package ## How was this patch tested? N/A Author: WeichenXu <WeichenXu123@outlook.com> Closes #14157 from WeichenXu123/add_lbfgs_convergence_warning_for_all_used_place.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala 5 additions, 0 deletions...g/apache/spark/ml/classification/LogisticRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala 5 additions, 0 deletions...rg/apache/spark/ml/regression/AFTSurvivalRegression.scala
- mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala 6 additions, 0 deletions...ain/scala/org/apache/spark/mllib/optimization/LBFGS.scala
Please register or sign in to comment