Skip to content

Commit be7ace0

Browse files
MechCodermarkhamstra
authored andcommitted
SPARK-4610 addendum: [Minor] [MLlib] Minor doc fix in GBT classification example
numClassesForClassification has been renamed to numClasses. Author: MechCoder <[email protected]> Closes apache#4672 from MechCoder/minor-doc and squashes the following commits: d2ddb7f [MechCoder] Minor doc fix in GBT classification example (cherry picked from commit e79a7a6) Signed-off-by: Sean Owen <[email protected]>
1 parent 4bfe8f1 commit be7ace0

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)