Skip to content

Commit d2ddb7f

Browse files
committed
Minor doc fix in GBT classification example
1 parent 3912d33 commit d2ddb7f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/mllib-ensembles.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -458,7 +458,7 @@ val (trainingData, testData) = (splits(0), splits(1))
458458
// The defaultParams for Classification use LogLoss by default.
459459
val boostingStrategy = BoostingStrategy.defaultParams("Classification")
460460
boostingStrategy.numIterations = 3 // Note: Use more iterations in practice.
461-
boostingStrategy.treeStrategy.numClassesForClassification = 2
461+
boostingStrategy.treeStrategy.numClasses = 2
462462
boostingStrategy.treeStrategy.maxDepth = 5
463463
// Empty categoricalFeaturesInfo indicates all features are continuous.
464464
boostingStrategy.treeStrategy.categoricalFeaturesInfo = Map[Int, Int]()

0 commit comments

Comments
 (0)