@@ -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