-
Notifications
You must be signed in to change notification settings - Fork 5
Closed
Labels
releasedIncluded in a releaseIncluded in a release
Description
Is your feature request related to a problem?
Decision trees are one of the few models that are easy to visualize. However, we don't offer any visualization at the moment.
Desired solution
- Add a method
plot() -> ImagetoDecisionTreeRegressorandDecisionTreeClassifier. - Raise a
ModelNotFittedErrorif the model is not fitted yet.
Implementation:
- Use
plot_tree. - Avoid code duplication, e.g. by adding the method to the common superclass
_DecisionTreeBase.
Possible alternatives (optional)
No response
Screenshots (optional)
No response
Additional Context (optional)
No response
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
releasedIncluded in a releaseIncluded in a release
Type
Projects
Status
✔️ Done