Skip to content

DOC Update plots in Categorical Feature Support in GBDT example #31062

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
wants to merge 5 commits into
base: main
Choose a base branch
from

Conversation

ArturoAmorQ
Copy link
Member

@ArturoAmorQ ArturoAmorQ commented Mar 24, 2025

Reference Issues/PRs

What does this implement/fix? Explain your changes.

Inspired by this example's way of showing the test_score/fit_time trade-off, I find a scatter plot easier to read and interpret than bar plots.

This PR also introduces a log scale for fit times.

Current plot:

sphx_glr_plot_gradient_boosting_categorical_main

This PR:

sphx_glr_plot_gradient_boosting_categorical_PR

Any other comments?

I took the opportunity to show the intermediate html diagrams and a introduce a wording tweak.

Copy link

github-actions bot commented Mar 24, 2025

✔️ Linting Passed

All linting checks passed. Your pull request is in excellent shape! ☀️

Generated for commit: c9771b9. Link to the linter CI: here

@ogrisel
Copy link
Member

ogrisel commented Apr 30, 2025

As discussed during the bi-weekly meeting, it would be great to make it explicit that the best models are in the bottom left corner, maybe using a matplotlib arrow annotation in the bottom left corner with the text "best models" pointing towards the point at coordinate (0, 0) (or (0.1, 0.1) because of the log scale on the x axis).

Also, could you please add more ticks on the x axis, e.g.: 0.1, 0.15, 0.2, 0.25, 0.3, 0.35, 0.4.

Also, please add the TargetEncoder to this plot.

@ArturoAmorQ
Copy link
Member Author

Current result:

image

Also, please add the TargetEncoder to this plot.

As that also requires adding narrative on the interpretation, I would rather leave that to a follow-up PR.

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

Successfully merging this pull request may close these issues.

2 participants