fix SparseVector.toArray

yinxusen · Mar 23, 2014 · 67b368d · 67b368d
1 parent 5eda0de
commit 67b368d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala
@@ -165,7 +165,8 @@ class SparseVector(val n: Int, val indices: Array[Int], val values: Array[Double
   override def toArray: Array[Double] = {
     val data = new Array[Double](n)
     var i = 0
-    while (i < n) {
+    val nnz = indices.length
+    while (i < nnz) {
       data(indices(i)) = values(i)
       i += 1
     }