[Minor][MLlib] Incorrect path to test data is used in DecisionTreeExa…

…mple It should load from `testInput` instead of `input` for test data. Author: Liang-Chi Hsieh <[email protected]> Closes apache#5594 from viirya/use_testinput and squashes the following commits: 5e8b174 [Liang-Chi Hsieh] Fix style. b60b475 [Liang-Chi Hsieh] Use testInput.
markhamstra · Apr 20, 2015 · 1ebceaa · 1ebceaa
1 parent 7717661
commit 1ebceaa
Showing 1 changed file with 2 additions 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,8 @@ 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.