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

Add more deps to cover whitebox TreeShap explainers #2935

Merged
merged 2 commits into from
Feb 8, 2021

Conversation

adriangonz
Copy link
Contributor

What this PR does / why we need it:

Add catboost and lightgbm as extra dependencies of the image. The overall size doesn't increase by that much (around 200MBs).

This PR also includes some changes that make the overall image a bit smaller than before.

Which issue(s) this PR fixes:

Fixes #2812

Special notes for your reviewer:

Note that this PR doesn't fully address the issue. As alibi evolves, the list of dependencies would need to keep growing. However, it's a simple enough short-term workaround which includes the most common GBT frameworks.

On the longer term, we are exploring a better solution from the MLServer side (SeldonIO/MLServer#123) which would allow users to specify custom environments on their explainers. This could be leveraged to match the training and production environments up to the exact library version.

Does this PR introduce a user-facing change?:

NONE

@adriangonz
Copy link
Contributor Author

/cc @cliveseldon @jklaise @RafalSkolasinski

@jklaise
Copy link
Contributor

jklaise commented Feb 8, 2021

/approve

LGTM. Agree with being more flexible in the future, even with different versions of the same libraries that users may want to specify instead.

@seldondev
Copy link
Collaborator

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: jklaise

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@seldondev seldondev merged commit 5305dee into SeldonIO:master Feb 8, 2021
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.

Alibi whitebox explainers unsupported due to missing dependencies in the Alibi explain server
3 participants