@@ -407,7 +407,7 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll {
407
407
408
408
val leftFilter = Filter (new Split (0 , 400 , FeatureType .Continuous , List ()), - 1 )
409
409
val rightFilter = Filter (new Split (0 , 400 , FeatureType .Continuous , List ()) ,1 )
410
- val filters = Array [List [Filter ]](List (),List (leftFilter), List (rightFilter))
410
+ val filters = Array [List [Filter ]](List (), List (leftFilter), List (rightFilter))
411
411
val parentImpurities = Array (0.5 , 0.5 , 0.5 )
412
412
413
413
// Single group second level tree construction.
@@ -463,7 +463,7 @@ object DecisionTreeSuite {
463
463
def generateOrderedLabeledPoints (): Array [LabeledPoint ] = {
464
464
val arr = new Array [LabeledPoint ](1000 )
465
465
for (i <- 0 until 1000 ) {
466
- if (i < 600 ){
466
+ if (i < 600 ) {
467
467
val lp = new LabeledPoint (0.0 , Vectors .dense(i.toDouble, 1000.0 - i))
468
468
arr(i) = lp
469
469
} else {
@@ -477,7 +477,7 @@ object DecisionTreeSuite {
477
477
def generateCategoricalDataPoints (): Array [LabeledPoint ] = {
478
478
val arr = new Array [LabeledPoint ](1000 )
479
479
for (i <- 0 until 1000 ) {
480
- if (i < 600 ){
480
+ if (i < 600 ) {
481
481
arr(i) = new LabeledPoint (1.0 , Vectors .dense(0.0 , 1.0 ))
482
482
} else {
483
483
arr(i) = new LabeledPoint (0.0 , Vectors .dense(1.0 , 0.0 ))
0 commit comments