aboutsummaryrefslogtreecommitdiff
path: root/mllib/src
diff options
context:
space:
mode:
authorReza Zadeh <rizlar@gmail.com>2014-09-02 09:48:05 -0700
committerXiangrui Meng <meng@databricks.com>2014-09-02 09:48:05 -0700
commit0f16b23cd17002fac05f3ecc58899be1b1121b82 (patch)
treed64f1e6c3bdc7e5985153ddd7447780b3c6f6e9d /mllib/src
parentfbf2678c16acc0071ebd1cbdd165702635be5f0c (diff)
downloadspark-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/src')
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/IndexedRowMatrix.scala2
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)
}
/**