Skip to content
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

Plotting a partial dependance corner plot (like scikit-optimize) #806

Open
avivajpeyi opened this issue Jan 11, 2024 · 0 comments
Open

Plotting a partial dependance corner plot (like scikit-optimize) #806

avivajpeyi opened this issue Jan 11, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@avivajpeyi
Copy link

Describe the feature you'd like
A potentially useful plot for diagnostics is the partial dependence 'corner' or 'triangle' plot of the model prediction in the various 1d/2d marginal views, as done in the scikit-optimize library

sphx_glr_partial-dependence-plot_001

This shows the partial dependence of the model output with the various model input parameters, and may help visualise/diagnose the progress the optimizer.

Describe alternatives you've considered
I've attempted at using corner.py to achieve the above in a hacky approach (by using the GP model mean value as 'weights'):

1. plot_samples = search_space.sample(N)
2. weights, _ = model.predict(plot_samples)
3. corner(plot_samples, weights=weights/max(weights))
Screenshot 2024-01-11 at 12 43 15 pm

this kindof works, but isn't the ideal solution -- the 2d and 1d contours can be made to be smooth if we plot the partial dependence of the model for each parameter rather than plotting samples weighted by the model output.

Would this be something that could be interesting to add as an 'experimental plottting' feature?

@uri-granta uri-granta added the enhancement New feature or request label Feb 4, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants