Github user BryanCutler commented on a diff in the pull request:
https://github.com/apache/spark/pull/10601#discussion_r51638874
--- Diff: python/pyspark/mllib/tree.py ---
@@ -385,30 +414,37 @@ def trainRegressor(cls, data,
categoricalFeaturesInfo, numTrees, featureSubsetSt
"""
Method to train a decision tree model for regression.
- :param data: Training dataset: RDD of LabeledPoint. Labels are
- real numbers.
- :param categoricalFeaturesInfo: Map storing arity of categorical
- features. E.g., an entry (n -> k) indicates that feature
- n is categorical with k categories indexed from 0:
- {0, 1, ..., k-1}.
- :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.
- :param impurity: Criterion used for information gain
- calculation.
- Supported values: "variance".
- :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: 32)
- :param seed: Random seed for bootstrapping and choosing feature
- subsets.
- :return: RandomForestModel that can be used for prediction
+ :param data:
+ Training dataset: RDD of LabeledPoint. Labels are real numbers.
+ :param categoricalFeaturesInfo:
+ Map storing arity of categorical features. E.g., an entry (n ->
+ k) indicates that feature n is categorical with k categories
+ indexed from 0: {0, 1, ..., k-1}.
+ :param numTrees:
+ Number of trees in the random forest.
+ :param featureSubsetStrategy:
+ Number of features to consider for splits at each node.
+ Supported: "auto", "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.
--- End diff --
Same as above for the `make html` warnings
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
