Github user MLnick commented on a diff in the pull request:
https://github.com/apache/spark/pull/11353#discussion_r54209067
--- Diff:
mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala ---
@@ -38,8 +38,9 @@ import org.apache.spark.util.random.XORShiftRandom
/**
* A class which implements a decision tree learning algorithm for
classification and regression.
* It supports both continuous and categorical features.
+ *
* @param strategy The configuration parameters for the tree algorithm
which specify the type
- * of algorithm (classification, regression, etc.),
feature type (continuous,
+ * of decision tree (classification or regression),
feature type (continuous,
* categorical), depth of the tree, quantile calculation
strategy, etc.
--- End diff --
`Strategy` includes the `categoricalFeaturesInfo` which I think is where
this part of the comment comes from.
It actually could be a little confusing, but I don't think it's a major
issue. It could read something like `which features are categorical` which
would be a little more accurate.
---
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]
