File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed
mllib/src/main/scala/org/apache/spark/mllib/tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change 11This package contains the default implementation of the decision tree algorithm.
22
33The decision tree algorithm supports:
4- + information loss calculation with entropy and gini for classification and variance for regression
5- + node model pruning
6- + printing to dot files
7- + unit tests
4+ + Binary classification
5+ + Regression
6+ + Information loss calculation with entropy and gini for classification and variance for regression
7+ + Both continuous and categorical features
88
9- #Performance testing
9+ # Tree improvements
10+ + Node model pruning
11+ + Printing to dot files
1012
11- #Future Extensions
13+ # Future Ensemble Extensions
1214
1315+ Random forests
1416+ Boosting
15- + Extremely randomized trees
17+ + Extremely randomized trees
You can’t perform that action at this time.
0 commit comments