Skip to content

Commit 7e5f08c

Browse files
committed
minor doc
1 parent 1dd2735 commit 7e5f08c

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -584,8 +584,7 @@ object DecisionTree extends Serializable with Logging {
584584
val binIndex = {
585585
if (strategy.isMultiClassification) {
586586
sequentialBinSearchForCategoricalFeatureInBinaryClassification()
587-
}
588-
else {
587+
} else {
589588
sequentialBinSearchForCategoricalFeatureInMultiClassClassification()
590589
}
591590
}
@@ -601,7 +600,7 @@ object DecisionTree extends Serializable with Logging {
601600
* For l nodes, k features the storage is as follows:
602601
* label, b_11, b_12, .. , b_1k, b_21, b_22, .. , b_2k, b_l1, b_l2, .. , b_lk,
603602
* where b_ij is an integer between 0 and numBins - 1 for regressions and binary
604-
* classification and an invalid value for categorical feature in multiclass classification.
603+
* classification and the categorical feature value in multiclass classification.
605604
* Invalid sample is denoted by noting bin for feature 1 as -1.
606605
*/
607606
def findBinsForLevel(labeledPoint: WeightedLabeledPoint): Array[Double] = {

0 commit comments

Comments
 (0)