From c1812a0a8a79629c16ff4cfe167c096f548a4764 Mon Sep 17 00:00:00 2001 From: lee19 Date: Tue, 30 Jun 2015 23:38:15 +0900 Subject: [PATCH] [SPARK-8563] [MLlib] Used nRows instead of numRows() to reduce a burden. --- .../spark/mllib/linalg/distributed/IndexedRowMatrix.scala | 2 +- 1 file changed, 1 insertion(+), 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 index b447441890d4a..1c33b43ea7a8a 100644 --- 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, svd.U.numRows(), svd.U.numCols().toInt) + new IndexedRowMatrix(indexedRows, nRows, svd.U.numCols().toInt) } else { null }