diff options
author | MechCoder <manojkumarsivaraj334@gmail.com> | 2015-02-18 10:13:28 +0000 |
---|---|---|
committer | Sean Owen <sowen@cloudera.com> | 2015-02-18 10:13:40 +0000 |
commit | e79a7a626d9ac2e2474b9d5008c6b5d07df5c6f1 (patch) | |
tree | 9bcdcf52ca464f0a9ed727b78712b7ff92cb4570 /docs/mllib-ensembles.md | |
parent | c1b6fa9838f9d26d60fab3b05a96649882e3dd5b (diff) | |
download | spark-e79a7a626d9ac2e2474b9d5008c6b5d07df5c6f1.tar.gz spark-e79a7a626d9ac2e2474b9d5008c6b5d07df5c6f1.tar.bz2 spark-e79a7a626d9ac2e2474b9d5008c6b5d07df5c6f1.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
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]() |