use to breeze for features

hhbyyh · hhbyyh · commit a709f49a751a · 2016-01-12T10:04:19.000+08:00
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/stat/test/ChiSqTest.scala b/mllib/src/main/scala/org/apache/spark/mllib/stat/test/ChiSqTest.scala
@@ -109,9 +109,9 @@ private[stat] object ChiSqTest extends Logging {
           }
           i += 1
           distinctLabels += label
-          val featureArray = features.toArray
+          val brzFeatures = features.toBreeze
           (startCol until endCol).map { col =>
-            val feature = featureArray(col)
+            val feature = brzFeatures(col)
             allDistinctFeatures(col) += feature
             (col, feature, label)
           }

Original file line number	Diff line number	Diff line change
`@@ -109,9 +109,9 @@ private[stat] object ChiSqTest extends Logging {`
`109`	`109`	`}`
`110`	`110`	`i += 1`
`111`	`111`	`distinctLabels += label`
`112`		`- val featureArray = features.toArray`
	`112`	`+ val brzFeatures = features.toBreeze`
`113`	`113`	`(startCol until endCol).map { col =>`
`114`		`- val feature = featureArray(col)`
	`114`	`+ val feature = brzFeatures(col)`
`115`	`115`	`allDistinctFeatures(col) += feature`
`116`	`116`	`(col, feature, label)`
`117`	`117`	`}`