diff options
| author | Joseph K. Bradley <joseph@databricks.com> | 2015-08-14 10:48:02 -0700 |
|---|---|---|
| committer | Joseph K. Bradley <joseph@databricks.com> | 2015-08-14 10:48:02 -0700 |
| commit | 7ecf0c46990c39df8aeddbd64ca33d01824bcc0a (patch) | |
| tree | b239fcb70b03a084323c76dd34c299107cd43181 /dev/lint-scala | |
| parent | a0e1abbd010b9e73d472ce12ff1d987678005d32 (diff) | |
| download | spark-7ecf0c46990c39df8aeddbd64ca33d01824bcc0a.tar.gz spark-7ecf0c46990c39df8aeddbd64ca33d01824bcc0a.tar.bz2 spark-7ecf0c46990c39df8aeddbd64ca33d01824bcc0a.zip | |
[SPARK-9956] [ML] Make trees work with one-category features
This modifies DecisionTreeMetadata construction to treat 1-category features as continuous, so that trees do not fail with such features. It is important for the pipelines API, where VectorIndexer can automatically categorize certain features as categorical.
As stated in the JIRA, this is a temp fix which we can improve upon later by automatically filtering out those features. That will take longer, though, since it will require careful indexing.
Targeted for 1.5 and master
CC: manishamde mengxr yanboliang
Author: Joseph K. Bradley <joseph@databricks.com>
Closes #8187 from jkbradley/tree-1cat.
Diffstat (limited to 'dev/lint-scala')
0 files changed, 0 insertions, 0 deletions