diff options
author | GuoQiang Li <witgo@qq.com> | 2014-11-16 21:31:51 -0800 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2014-11-16 21:31:51 -0800 |
commit | 5168c6ca9f0008027d688661bae57c28cf386b54 (patch) | |
tree | 8a540d39bca6393b22f74265057162408315d1ff /mllib/src/main | |
parent | 45ce3273cb618d14ec4d20c4c95699634b951086 (diff) | |
download | spark-5168c6ca9f0008027d688661bae57c28cf386b54.tar.gz spark-5168c6ca9f0008027d688661bae57c28cf386b54.tar.bz2 spark-5168c6ca9f0008027d688661bae57c28cf386b54.zip |
[SPARK-4422][MLLIB]In some cases, Vectors.fromBreeze get wrong results.
cc mengxr
Author: GuoQiang Li <witgo@qq.com>
Closes #3281 from witgo/SPARK-4422 and squashes the following commits:
5f1fa5e [GuoQiang Li] import order
50783bd [GuoQiang Li] review commits
7a10123 [GuoQiang Li] In some cases, Vectors.fromBreeze get wrong results.
Diffstat (limited to 'mllib/src/main')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala index 9fccd6341b..60ab2aaa8f 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala @@ -237,7 +237,7 @@ object Vectors { private[mllib] def fromBreeze(breezeVector: BV[Double]): Vector = { breezeVector match { case v: BDV[Double] => - if (v.offset == 0 && v.stride == 1) { + if (v.offset == 0 && v.stride == 1 && v.length == v.data.length) { new DenseVector(v.data) } else { new DenseVector(v.toArray) // Can't use underlying array directly, so make a new one |