Skip to content

Commit

Permalink
basic doc
Browse files Browse the repository at this point in the history
Signed-off-by: Manish Amde <manish9ue@gmail.com>
  • Loading branch information
manishamde committed Feb 28, 2014
1 parent 5841c28 commit 0dd7659
Showing 1 changed file with 9 additions and 7 deletions.
16 changes: 9 additions & 7 deletions mllib/src/main/scala/org/apache/spark/mllib/tree/README.md
Original file line number Diff line number Diff line change
@@ -1,15 +1,17 @@
This package contains the default implementation of the decision tree algorithm.

The decision tree algorithm supports:
+ information loss calculation with entropy and gini for classification and variance for regression
+ node model pruning
+ printing to dot files
+ unit tests
+ Binary classification
+ Regression
+ Information loss calculation with entropy and gini for classification and variance for regression
+ Both continuous and categorical features

#Performance testing
# Tree improvements
+ Node model pruning
+ Printing to dot files

#Future Extensions
# Future Ensemble Extensions

+ Random forests
+ Boosting
+ Extremely randomized trees
+ Extremely randomized trees

0 comments on commit 0dd7659

Please sign in to comment.