Bug fix of sparse vector conversion

funes · funes · commit b85806c190f8 · 2014-05-06T11:49:52.000+08:00
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
@@ -136,7 +136,7 @@ object Vectors {
           new DenseVector(v.toArray)  // Can't use underlying array directly, so make a new one
         }
       case v: BSV[Double] =>
-        new SparseVector(v.length, v.index, v.data)
+        new SparseVector(v.length, v.index.slice(0, v.used), v.data.slice(0, v.used))
       case v: BV[_] =>
         sys.error("Unsupported Breeze vector type: " + v.getClass.getName)
     }

Original file line number	Diff line number	Diff line change
`@@ -136,7 +136,7 @@ object Vectors {`
`136`	`136`	`new DenseVector(v.toArray) // Can't use underlying array directly, so make a new one`
`137`	`137`	`}`
`138`	`138`	`case v: BSV[Double] =>`
`139`		`- new SparseVector(v.length, v.index, v.data)`
	`139`	`+ new SparseVector(v.length, v.index.slice(0, v.used), v.data.slice(0, v.used))`
`140`	`140`	`case v: BV[_] =>`
`141`	`141`	`sys.error("Unsupported Breeze vector type: " + v.getClass.getName)`
`142`	`142`	`}`