diff options
Diffstat (limited to 'mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java')
| -rw-r--r-- | mllib/src/test/java/org/apache/spark/mllib/tree/JavaDecisionTreeSuite.java | 4 |
1 files changed, 2 insertions, 2 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 9925aae441..8dd29061da 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 @@ -64,7 +64,7 @@ public class JavaDecisionTreeSuite implements Serializable { public void runDTUsingConstructor() { List<LabeledPoint> arr = DecisionTreeSuite.generateCategoricalDataPointsAsJavaList(); JavaRDD<LabeledPoint> rdd = sc.parallelize(arr); - HashMap<Integer, Integer> categoricalFeaturesInfo = new HashMap<Integer, Integer>(); + HashMap<Integer, Integer> categoricalFeaturesInfo = new HashMap<>(); categoricalFeaturesInfo.put(1, 2); // feature 1 has 2 categories int maxDepth = 4; @@ -84,7 +84,7 @@ public class JavaDecisionTreeSuite implements Serializable { public void runDTUsingStaticMethods() { List<LabeledPoint> arr = DecisionTreeSuite.generateCategoricalDataPointsAsJavaList(); JavaRDD<LabeledPoint> rdd = sc.parallelize(arr); - HashMap<Integer, Integer> categoricalFeaturesInfo = new HashMap<Integer, Integer>(); + HashMap<Integer, Integer> categoricalFeaturesInfo = new HashMap<>(); categoricalFeaturesInfo.put(1, 2); // feature 1 has 2 categories int maxDepth = 4; |