Skip to content

Commit 1b29c13

Browse files
committed
Merge branch 'decisiontree-bugfix' into decisiontree-python-new
2 parents 584449a + dab0b67 commit 1b29c13

5 files changed

Lines changed: 227 additions & 108 deletions

File tree

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)