diff options
| author | Joseph K. Bradley <joseph@databricks.com> | 2015-04-01 15:15:47 -0700 |
|---|---|---|
| committer | Reynold Xin <rxin@databricks.com> | 2015-04-01 15:15:47 -0700 |
| commit | fb25e8c7f45b4f96561e3f7434a0f4dfce8ddefe (patch) | |
| tree | 7829d3966638885088bc6daf9d388979d91a8363 /python/pyspark/mllib | |
| parent | 2275acce7ba5fac83c58554d7ee9f4c7f3e866cf (diff) | |
| download | spark-fb25e8c7f45b4f96561e3f7434a0f4dfce8ddefe.tar.gz spark-fb25e8c7f45b4f96561e3f7434a0f4dfce8ddefe.tar.bz2 spark-fb25e8c7f45b4f96561e3f7434a0f4dfce8ddefe.zip | |
[SPARK-6657] [Python] [Docs] fixed python doc build warnings
fixed python doc build warnings
CC whomever wants to review: rxin mengxr davies
Author: Joseph K. Bradley <joseph@databricks.com>
Closes #5317 from jkbradley/python-doc-warnings and squashes the following commits:
4cd43c2 [Joseph K. Bradley] fixed python doc build warnings
Diffstat (limited to 'python/pyspark/mllib')
| -rw-r--r-- | python/pyspark/mllib/tree.py | 26 |
1 files changed, 10 insertions, 16 deletions
diff --git a/python/pyspark/mllib/tree.py b/python/pyspark/mllib/tree.py index bf288d7644..a7a4d2aaf8 100644 --- a/python/pyspark/mllib/tree.py +++ b/python/pyspark/mllib/tree.py @@ -286,21 +286,18 @@ class RandomForest(object): :param numTrees: Number of trees in the random forest. :param featureSubsetStrategy: Number of features to consider for splits at each node. - Supported: "auto" (default), "all", "sqrt", "log2", - "onethird". - If "auto" is set, this parameter is set based on - numTrees: - if numTrees == 1, set to "all"; - if numTrees > 1 (forest) set to "sqrt". - :param impurity: Criterion used for information gain - calculation. + Supported: "auto" (default), "all", "sqrt", "log2", "onethird". + If "auto" is set, this parameter is set based on numTrees: + if numTrees == 1, set to "all"; + if numTrees > 1 (forest) set to "sqrt". + :param impurity: Criterion used for information gain calculation. Supported values: "gini" (recommended) or "entropy". :param maxDepth: Maximum depth of the tree. E.g., depth 0 means 1 leaf node; depth 1 means 1 internal node + 2 leaf nodes. (default: 4) :param maxBins: maximum number of bins used for splitting features - (default: 100) + (default: 100) :param seed: Random seed for bootstrapping and choosing feature subsets. :return: RandomForestModel that can be used for prediction @@ -365,13 +362,10 @@ class RandomForest(object): :param numTrees: Number of trees in the random forest. :param featureSubsetStrategy: Number of features to consider for splits at each node. - Supported: "auto" (default), "all", "sqrt", "log2", - "onethird". - If "auto" is set, this parameter is set based on - numTrees: - if numTrees == 1, set to "all"; - if numTrees > 1 (forest) set to "onethird" for - regression. + Supported: "auto" (default), "all", "sqrt", "log2", "onethird". + If "auto" is set, this parameter is set based on numTrees: + if numTrees == 1, set to "all"; + if numTrees > 1 (forest) set to "onethird" for regression. :param impurity: Criterion used for information gain calculation. Supported values: "variance". |