Skip to content

Commit 68fde20

Browse files
committed
Merge pull request scikit-learn#3735 from trevorstephens/pretty-decision-trees
[MRG] Pretty decision trees - tree.export_graphviz()
2 parents f65d7a7 + 2718499 commit 68fde20

File tree

3 files changed

+484
-60
lines changed

3 files changed

+484
-60
lines changed

doc/whats_new.rst

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,12 @@ Enhancements
4242
- Added ``stratify`` option to :func:`cross_validation.train_test_split` for stratified splitting.
4343
By Miroslav Batchkarov.
4444

45+
- The :func:`tree.export_graphviz` function now supports aesthetic
46+
improvements for :class:`tree.DecisionTreeClassifier` and
47+
:class:`tree.DecisionTreeRegressor`, including options for coloring nodes
48+
by their majority class or impurity, showing variable names, and using
49+
node proportions instead of raw sample counts. By `Trevor Stephens`_.
50+
4551
Bug fixes
4652
.........
4753

0 commit comments

Comments
 (0)