This is a visualization of a decision tree and the associated decision boundary for the binary classifier. The tree visualizer uses graphviz and the decision boundary uses a function based on matplotlib's pyplot. The well-known make_moons dataset is used, and a decision tree model is trained on it (including hyperparamter tuning). The decision tree boundary is compared to that of a tree with default choices of parameters (which results in overfitting on this dataset).
-
Notifications
You must be signed in to change notification settings - Fork 0
sanjayc2/Decision_Tree_Visualization
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
This is a visualization of a decision tree and its decision boundary.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published