mathias-brandewinder / charon Goto Github PK

When the tree has no features as input, or no feature is deemed informative by the tree analysis, the "Pretty" tree shows just the label name, but not the default tree prediction.

The current Pretty Tree sometimes keeps searching and rendering branches, although every single Leaf (for instance) has the same default outcome [see for instance the Tree obtained on Titanic with Sex, Class].
Having a "compact" Tree pruning the "uninformative" branches (i.e. branches where every single Leaf has the same prediction) would make for a more readable output.

Current Pretty Tree simply produces the branch and the most likely outcome. It would be nice to add the probability of the branch and/or outcome, too, ex:

• Male (0.430) -> Survived : false (0.731)

It might be useful to have a simple compact tree, but also the option to produce a full tree, ex: adding every single outcome with their probability in each branch.

Add unit test for commit 714f38a