Skip to content

Commit dab0b67

Browse files
committed
Added documentation for DecisionTree internals
1 parent 8bb8aa0 commit dab0b67

File tree

3 files changed

+199
-84
lines changed

3 files changed

+199
-84
lines changed

examples/src/main/scala/org/apache/spark/examples/mllib/DecisionTreeRunner.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ object DecisionTreeRunner {
145145
println(s"Per-class example fractions, counts:")
146146
println(s"Class\tFrac\tCount")
147147
sortedClasses.foreach { c => {
148-
val frac = classCounts(c) / (0.0 + numExamples)
148+
val frac = classCounts(c) / numExamples.toDouble
149149
println(s"$c\t$frac\t${classCounts(c)}")
150150
}}
151151
(examples, numClasses)

0 commit comments

Comments
 (0)