Open OlivierBinette opened 1 year ago
It might be a good idea to follow the same interface as the plot_tree
function from scikit-learn to start: https://scikit-learn.org/stable/modules/generated/sklearn.tree.plot_tree.html
[Work in progress] Based on the "wants" discusses in Issue#2, include the following functions:
plot_tree
: takes in the fitted classifier and plotting arguments; returns a tree plot (matplotlib
figure)export_tree
: takes in the fitted classifier; returns a tree of text form (string
)plot_decision_boundary
: takes in the fitted classifier and plotting arguments; returns boundary plots for each pair of features
Any ideas for functions to include on the package? What should be the functions signatures, return type and behaviors?
For reference, here is code to plot decision trees using plotly: https://github.com/OlivierBinette/er-evaluation/blob/main/er_evaluation/plots/_dtree_plots.py. The results can be seen at the bottom of the page here: https://er-evaluation.readthedocs.io/en/latest/visualizations.html