Skip to content

Adding more functionality to DecisionBoundaryDisplay #27462

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
2 of 5 tasks
glemaitre opened this issue Sep 25, 2023 · 2 comments
Open
2 of 5 tasks

Adding more functionality to DecisionBoundaryDisplay #27462

glemaitre opened this issue Sep 25, 2023 · 2 comments
Assignees

Comments

@glemaitre
Copy link
Member

glemaitre commented Sep 25, 2023

This issue is to track the improvements that should be made to the DecisionBoundaryDisplay:

@StefanieSenger
Copy link
Contributor

I have used DecisionBoundaryDisplay to plot from_estimator of a DecisionTreeClassifier as in this example.

Based on that I would suggest further improvements:

  1. add a title to the plot that clearly states which response_method from which estimator was used in the plot
  2. add a legend that shows the values (or value regions) of the response method used (percentages for predict_proba, classes for proba, raw values for decision_function)
    • 2b. redefine the representation of the legend that defines the classes, since they look like a scale rather than marking membership

Do you think this would be useful, @glemaitre?

@glemaitre
Copy link
Member Author

I agree that we can improve the plot. Maybe some of those improvement would be opt-in via parameter. I need to think more about each of them but having proof of concept will help anyway to know if we want it a certain way.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Development

No branches or pull requests

2 participants