diff options
Diffstat (limited to 'mllib/src/test')
| -rw-r--r-- | mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java b/mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java index 0f71deb9ea..d2fe6bb2ca 100644 --- a/mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java +++ b/mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java @@ -33,7 +33,8 @@ import org.apache.spark.mllib.tree.model.DecisionTreeModel; public class JavaDecisionTreeSuite extends SharedSparkSession { - private static int validatePrediction(List<LabeledPoint> validationData, DecisionTreeModel model) { + private static int validatePrediction( + List<LabeledPoint> validationData, DecisionTreeModel model) { int numCorrect = 0; for (LabeledPoint point : validationData) { Double prediction = model.predict(point.features()); |