minor fix: remove empty lines

apache · Sep 9, 2014 · 6728fad · 6728fad
1 parent bb465ca
commit 6728fad
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
@@ -747,7 +747,6 @@ object DecisionTree extends Serializable with Logging {
 
     val totalCount = leftCount + rightCount
 
-
     // impurity of parent node
     val impurity = if (level > 0) {
       topImpurity
@@ -836,7 +835,7 @@ object DecisionTree extends Serializable with Logging {
               calculateGainForSplit(leftChildStats, rightChildStats, nodeImpurity, level, metadata)
             (splitIdx, gainStats)
           }.maxBy(_._2.gain)
-        if (bestFeatureGainStats.gain < metadata.minInfoGain) {
+        if (bestFeatureGainStats.gain == metadata.minInfoGain) {
           (Split.noSplit, InformationGainStats.invalidInformationGainStats)
         } else {
           (splits(featureIndex)(bestFeatureSplitIndex), bestFeatureGainStats)