From 4f22c2bf59527eeb49487ff5bd0a869cade4f251 Mon Sep 17 00:00:00 2001 From: lewuathe Date: Wed, 1 Jul 2015 21:47:11 +0900 Subject: [PATCH] Modify based on SPARK-1503 --- .../org/apache/spark/mllib/optimization/GradientDescent.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala b/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala index 31491afb0db10..4df861f8045e5 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala @@ -291,7 +291,7 @@ object GradientDescent extends Logging { // This represents the difference of updated weights in the iteration. val solutionVecDiff: Double = norm(previousBDV - currentBDV) - solutionVecDiff < convergenceTol * norm(previousBDV) + solutionVecDiff < convergenceTol * Math.max(norm(currentBDV), 1.0) } }