Tabpfn explainer #302
Merged
Tabpfn explainer #302
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
mmschlk
requested review from
hbaniecki and
FFmgll
and removed request for
hbaniecki
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request introduces a new explainer class,
TabPFNExplainer, and its associated imputer,TabPFNImputer, to theshapiqpackage. These changes include updates to the initialization and validation processes, as well as modifications to existing classes to accommodate the new explainer. The most important changes include the addition of the new explainer and imputer, updates to theExplainerclass initialization, and improvements to data validation.New Explainer and Imputer:
shapiq/explainer/tabpfn.py: Added theTabPFNExplainerclass for explaining the predictions of TabPFN models.shapiq/games/imputer/tabpfn_imputer.py: Added theTabPFNImputerclass to support the newTabPFNExplainer.Updates to
ExplainerClass:shapiq/explainer/_base.py: Modified theExplainerclass to include the_validate_datamethod for data validation and updated the initialization process to dynamically set the explainer class based on the model type. [1] [2]_prediction_functionto_shapiq_predict_functionsuch that the prediction function which shapiq may add to objects during explanations does not clash and overrides potential_predict_functionfunctionsIntegration of New Explainer and Imputer:
shapiq/__init__.py: Updated imports and__all__to includeTabPFNExplainerandTabPFNImputer. [1] [2]shapiq/explainer/__init__.py: AddedTabPFNExplainerto the module exports.shapiq/games/__init__.py: Updated imports and__all__to includeTabPFNImputer.