ODIn/spark-instrumented-optimizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

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

Browse source 
It should load from `testInput` instead of `input` for test data.

Author: Liang-Chi Hsieh <viirya@gmail.com>

Closes #5594 from viirya/use_testinput and squashes the following commits:

5e8b174 [Liang-Chi Hsieh] Fix style.
b60b475 [Liang-Chi Hsieh] Use testInput.
This commit is contained in:
Liang-Chi Hsieh 2015-04-20 10:47:37 -07:00 committed by Joseph K. Bradley
parent 77176619a9
commit 1ebceaa55b
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
examples/src/main/scala/org/apache/spark/examples/ml/DecisionTreeExample.scala
View file

@ -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.