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

Lookup Table for single feature and feature interaction terms #530

Open
sauvikd opened this issue Apr 19, 2024 · 1 comment
Open

Lookup Table for single feature and feature interaction terms #530

sauvikd opened this issue Apr 19, 2024 · 1 comment

Comments

@sauvikd
Copy link

sauvikd commented Apr 19, 2024

Screenshot 2024-04-19 035202

How can I extract a lookup table for the red-line (score values for the each of the features) and similarly the data for the heatmap.

I had tried downloading the ebm_global.data(1)['scores'], but those are values for each of the bins, from what I understand. Even if I'm able to extract the bin boundary values, I will be able to connect them with the feature values.

@paulbkoch
Copy link
Collaborator

Hi @sauvikd --

One of the very nice things about EBMs, and GAMs in general, is that the scores are used when calculating the predictions AND what are shown on the graphs. EBMs have no hidden information.

For graphing though, you also need the "bins" attribute, which for continuous features are the locations on the X-axis where the bin transitions occur.

There's some code which creates graphs here: #325

And you might also find our documentation on EBM internals useful: https://interpret.ml/docs/ebm-internals-regression.html

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

No branches or pull requests

2 participants