Modify based on SPARK-1503

maropu · Jul 1, 2015 · 4f22c2b · 4f22c2b
1 parent 5e47b82
commit 4f22c2b
Showing 1 changed file with 1 addition and 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
@@ -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)
   }
 
 }