Skip to content

MultiLabel Classification Explainer#79

Merged
cdpierse merged 15 commits intomasterfrom
feature/multilabel-classification-explainer
Mar 3, 2022
Merged

MultiLabel Classification Explainer#79
cdpierse merged 15 commits intomasterfrom
feature/multilabel-classification-explainer

Conversation

@cdpierse
Copy link
Owner

@cdpierse cdpierse commented Mar 3, 2022

PR Description

Extends the existing sequence classification explainer into a new explainer that independently produces attributions for each label in the model regardless of what the predicted class is. This allows users to better inspect and interpret model predictions across all classes, particularly in situations where classifiers might be used in a multilabel fashion.

The MultiLabelClassificationExplainer returns a dictionary mapping labels/classes to a list of word attributions, additionally the visualize() method will display the entire table of attributions for each label.

This has been a very requested feature for a number of months so we're very happy to get it released (finally)

CC: @MichalMalyska @rhettdsouza13 @fraserprice @JensVN98 @dheerajiiitv

Motivation and Context

References issue: #39

Tests and Coverage

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Docs (Added to or improved Transformers Interpret's documentation)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)

Final Checklist:

  • My code follows the code style of this project.
  • I have updated the documentation accordingly.
  • I have added tests to cover my changes.
  • All new and existing tests passed.

@cdpierse cdpierse added the enhancement New feature or request label Mar 3, 2022
@cdpierse cdpierse self-assigned this Mar 3, 2022
@cdpierse cdpierse merged commit eb67078 into master Mar 3, 2022
@cdpierse cdpierse deleted the feature/multilabel-classification-explainer branch September 15, 2022 11:41
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

Successfully merging this pull request may close these issues.

1 participant

Comments