Skip to content

Commit eddac63

Browse files
committed
cleanup
1 parent 3bb28fe commit eddac63

File tree

1 file changed

+12
-17
lines changed

1 file changed

+12
-17
lines changed

mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala

Lines changed: 12 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -718,23 +718,18 @@ private[spark] object RandomForest extends Logging {
718718
splitIndex += 1
719719
}
720720
// Find best split.
721-
if (numSplits == 0 && false) {
722-
(new ContinuousSplit(featureIndex, Double.MinValue),
723-
ImpurityStats.getInvalidImpurityStats(binAggregates.getParentImpurityCalculator()))
724-
} else {
725-
val (bestFeatureSplitIndex, bestFeatureGainStats) =
726-
Range(0, numSplits).map { case splitIdx =>
727-
val leftChildStats =
728-
binAggregates.getImpurityCalculator(nodeFeatureOffset, splitIdx)
729-
val rightChildStats =
730-
binAggregates.getImpurityCalculator(nodeFeatureOffset, numSplits)
731-
rightChildStats.subtract(leftChildStats)
732-
gainAndImpurityStats = calculateImpurityStats(gainAndImpurityStats,
733-
leftChildStats, rightChildStats, binAggregates.metadata)
734-
(splitIdx, gainAndImpurityStats)
735-
}.maxBy(_._2.gain)
736-
(splits(featureIndex)(bestFeatureSplitIndex), bestFeatureGainStats)
737-
}
721+
val (bestFeatureSplitIndex, bestFeatureGainStats) =
722+
Range(0, numSplits).map { case splitIdx =>
723+
val leftChildStats =
724+
binAggregates.getImpurityCalculator(nodeFeatureOffset, splitIdx)
725+
val rightChildStats =
726+
binAggregates.getImpurityCalculator(nodeFeatureOffset, numSplits)
727+
rightChildStats.subtract(leftChildStats)
728+
gainAndImpurityStats = calculateImpurityStats(gainAndImpurityStats,
729+
leftChildStats, rightChildStats, binAggregates.metadata)
730+
(splitIdx, gainAndImpurityStats)
731+
}.maxBy(_._2.gain)
732+
(splits(featureIndex)(bestFeatureSplitIndex), bestFeatureGainStats)
738733
} else if (binAggregates.metadata.isUnordered(featureIndex)) {
739734
// Unordered categorical feature
740735
val leftChildOffset = binAggregates.getFeatureOffset(featureIndexIdx)

0 commit comments

Comments
 (0)