From 0f16b23cd17002fac05f3ecc58899be1b1121b82 Mon Sep 17 00:00:00 2001
From: Reza Zadeh <rizlar@gmail.com>
Date: Tue, 2 Sep 2014 09:48:05 -0700
Subject: [PATCH] [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
---
.../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 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)
}
/**
--
GitLab