Skip to content

Commit efcc736

Browse files
author
qiping.lqp
committed
fix bug
1 parent 10b8012 commit efcc736

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -835,7 +835,7 @@ object DecisionTree extends Serializable with Logging {
835835
calculateGainForSplit(leftChildStats, rightChildStats, nodeImpurity, level, metadata)
836836
(splitIdx, gainStats)
837837
}.maxBy(_._2.gain)
838-
if (bestFeatureGainStats.gain == metadata.minInfoGain) {
838+
if (bestFeatureGainStats == InformationGainStats.invalidInformationGainStats) {
839839
(Split.noSplit, InformationGainStats.invalidInformationGainStats)
840840
} else {
841841
(splits(featureIndex)(bestFeatureSplitIndex), bestFeatureGainStats)

0 commit comments

Comments
 (0)