fix bug

apache · Sep 9, 2014 · efcc736 · efcc736
1 parent 10b8012
commit efcc736
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -835,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 == InformationGainStats.invalidInformationGainStats) {
           (Split.noSplit, InformationGainStats.invalidInformationGainStats)
         } else {
           (splits(featureIndex)(bestFeatureSplitIndex), bestFeatureGainStats)