Use testInput.

apache · Apr 20, 2015 · b60b475 · b60b475
1 parent 6fe690d
commit b60b475
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/src/main/scala/org/apache/spark/examples/ml/DecisionTreeExample.scala b/examples/src/main/scala/org/apache/spark/examples/ml/DecisionTreeExample.scala
@@ -173,7 +173,7 @@ object DecisionTreeExample {
     val splits: Array[RDD[LabeledPoint]] = if (testInput != "") {
       // Load testInput.
       val numFeatures = origExamples.take(1)(0).features.size
-      val origTestExamples: RDD[LabeledPoint] = loadData(sc, input, dataFormat, Some(numFeatures))
+      val origTestExamples: RDD[LabeledPoint] = loadData(sc, testInput, dataFormat, Some(numFeatures))
       Array(origExamples, origTestExamples)
     } else {
       // Split input into training, test.