diff options
author | Reza Zadeh <rizlar@gmail.com> | 2014-09-02 09:48:05 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2014-09-02 09:48:05 -0700 |
commit | 0f16b23cd17002fac05f3ecc58899be1b1121b82 (patch) | |
tree | d64f1e6c3bdc7e5985153ddd7447780b3c6f6e9d /mllib | |
parent | fbf2678c16acc0071ebd1cbdd165702635be5f0c (diff) | |
download | spark-0f16b23cd17002fac05f3ecc58899be1b1121b82.tar.gz spark-0f16b23cd17002fac05f3ecc58899be1b1121b82.tar.bz2 spark-0f16b23cd17002fac05f3ecc58899be1b1121b82.zip |
[MLlib] Squash bug in IndexedRowMatrix
Kill this bug fast before it does damage.
Author: Reza Zadeh <rizlar@gmail.com>
Closes #2224 from rezazadeh/indexrmbug and squashes the following commits:
53386d6 [Reza Zadeh] Squash bug in IndexedRowMatrix
Diffstat (limited to 'mllib')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala | 2 |
1 files changed, 1 insertions, 1 deletions
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 132b3af72d..ac6eaea3f4 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 @@ -130,7 +130,7 @@ class IndexedRowMatrix( val indexedRows = rows.map(_.index).zip(mat.rows).map { case (i, v) => IndexedRow(i, v) } - new IndexedRowMatrix(indexedRows, nRows, nCols) + new IndexedRowMatrix(indexedRows, nRows, B.numCols) } /** |