diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala index 859f4f72957c7..8e7a6917946b8 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala @@ -292,8 +292,7 @@ object DecisionTree extends Serializable with Logging { maxDepth: Int, numClassesForClassification: Int, labelWeights: Map[Int,Int]): DecisionTreeModel = { - val strategy - = new Strategy(algo, impurity, maxDepth, numClassesForClassification, + val strategy = new Strategy(algo, impurity, maxDepth, numClassesForClassification, labelWeights = labelWeights) // Converting from standard instance format to weighted input format for tree training val weightedInput = input.map(x => WeightedLabeledPoint(x.label, x.features))