[SPARK-8563][MLlib] Fixed a bug so that IndexedRowMatrix.computeSVD()…

….U.numCols = k
apache · Jun 23, 2015 · 8373424 · 8373424
1 parent f0dcbe8
commit 8373424
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala
@@ -146,7 +146,7 @@ class IndexedRowMatrix(
       val indexedRows = indices.zip(svd.U.rows).map { case (i, v) =>
         IndexedRow(i, v)
       }
-      new IndexedRowMatrix(indexedRows, nRows, nCols)
+      new IndexedRowMatrix(indexedRows, svd.U.numRows(), svd.U.numCols())
     } else {
       null
     }