Skip to content

Commit e3c84cc

Browse files
committed
Added stuff fro mnist8m to D T Runner
1 parent 51ef781 commit e3c84cc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,10 @@ object DecisionTreeRunner {
156156
throw new IllegalArgumentException("Algo ${params.algo} not supported.")
157157
}
158158

159+
println("opt3")
159160
// Split into training, test.
160-
val splits = examples.randomSplit(Array(1.0 - params.fracTest, params.fracTest))
161+
val splitsTMP = examples.randomSplit(Array(0.9, 0.1), seed = 1234)
162+
val splits = splitsTMP(1).randomSplit(Array(1.0 - params.fracTest, params.fracTest), seed = 12345)
161163
val training = splits(0).cache()
162164
val test = splits(1).cache()
163165
val numTraining = training.count()

0 commit comments

Comments
 (0)