diff options
author | MechCoder <manojkumarsivaraj334@gmail.com> | 2015-02-18 10:13:28 +0000 |
---|---|---|
committer | Sean Owen <sowen@cloudera.com> | 2015-02-18 10:14:18 +0000 |
commit | 3997e74f6f60ed88b84b0a7cdb4d29a05f156b51 (patch) | |
tree | 72633f690cde7977cbcab5588d5234372156e2b3 /docs/mllib-ensembles.md | |
parent | aca7991597ce655fc18cdf090b4d087e86f93ffe (diff) | |
download | spark-3997e74f6f60ed88b84b0a7cdb4d29a05f156b51.tar.gz spark-3997e74f6f60ed88b84b0a7cdb4d29a05f156b51.tar.bz2 spark-3997e74f6f60ed88b84b0a7cdb4d29a05f156b51.zip |
SPARK-4610 addendum: [Minor] [MLlib] Minor doc fix in GBT classification example
numClassesForClassification has been renamed to numClasses.
Author: MechCoder <manojkumarsivaraj334@gmail.com>
Closes #4672 from MechCoder/minor-doc and squashes the following commits:
d2ddb7f [MechCoder] Minor doc fix in GBT classification example
(cherry picked from commit e79a7a626d9ac2e2474b9d5008c6b5d07df5c6f1)
Signed-off-by: Sean Owen <sowen@cloudera.com>
Diffstat (limited to 'docs/mllib-ensembles.md')
-rw-r--r-- | docs/mllib-ensembles.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/docs/mllib-ensembles.md b/docs/mllib-ensembles.md index 23ede04b62..fb90b70399 100644 --- a/docs/mllib-ensembles.md +++ b/docs/mllib-ensembles.md @@ -458,7 +458,7 @@ val (trainingData, testData) = (splits(0), splits(1)) // The defaultParams for Classification use LogLoss by default. val boostingStrategy = BoostingStrategy.defaultParams("Classification") boostingStrategy.numIterations = 3 // Note: Use more iterations in practice. -boostingStrategy.treeStrategy.numClassesForClassification = 2 +boostingStrategy.treeStrategy.numClasses = 2 boostingStrategy.treeStrategy.maxDepth = 5 // Empty categoricalFeaturesInfo indicates all features are continuous. boostingStrategy.treeStrategy.categoricalFeaturesInfo = Map[Int, Int]() |