Skip to content

Commit

Permalink
minor: another minor code style
Browse files Browse the repository at this point in the history
  • Loading branch information
manishamde committed May 26, 2014
1 parent 34ee7b9 commit 23d4268
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down

0 comments on commit 23d4268

Please sign in to comment.