Skip to content

Commit

Permalink
Merge branch 'dt-findsplits' of https://github.com/chouqin/spark into…
Browse files Browse the repository at this point in the history
… dt-findsplits

Conflicts:
	mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
  • Loading branch information
chouqin committed Oct 13, 2014
2 parents 1b25a35 + 0cd744a commit 9e7138e
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -917,9 +917,9 @@ object DecisionTree extends Serializable with Logging {
val featureSamples = sampledInput.map(lp => lp.features(featureIndex)).sorted
val featureSplits = findSplits(featureSamples, metadata.numSplits(featureIndex))
val numSplits = featureSplits.length
logDebug("numSplits= " + numSplits)
metadata.setNumBinForFeature(featureIndex, numSplits + 1)
val numBins = metadata.numBins(featureIndex)
logDebug("numSplits= " + numSplits)

splits(featureIndex) = new Array[Split](numSplits)
bins(featureIndex) = new Array[Bin](numBins)
Expand Down

0 comments on commit 9e7138e

Please sign in to comment.