Skip to content

Commit 438a660

Browse files
committed
removed subsampling for mnist8m from DT
1 parent 86e217f commit 438a660

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,8 +158,7 @@ object DecisionTreeRunner {
158158

159159
println("opt3")
160160
// Split into training, test.
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)
161+
val splits = examples.randomSplit(Array(1.0 - params.fracTest, params.fracTest), seed = 12345)
163162
val training = splits(0).cache()
164163
val test = splits(1).cache()
165164
val numTraining = training.count()

0 commit comments

Comments
 (0)