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

Fix MultiDimensional attribution aggregation #292

Merged
merged 3 commits into from
Nov 10, 2024
Merged

Conversation

gsarti
Copy link
Member

@gsarti gsarti commented Nov 8, 2024

Description

This PR fixes the issue highlighted in #291 in which the information contained in configuration attributes (preceded by _) in FeatureAttributionSequenceOutput classes was not copied over after calling out.aggregate. This was causing methods like value_zeroing to break, since the default MultiDimensionalFeatureAttributionSequenceOutput output was assumed to be 4-dimensional from the attention method.

In the updated version, a new property config was added to FeatureAttributionSequenceOutput to easily access all configuration attributes, and the resulting properties are now updated correctly.

@gsarti gsarti merged commit 1078e19 into main Nov 10, 2024
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant