Skip to content

Commit

Permalink
Updated docs
Browse files Browse the repository at this point in the history
  • Loading branch information
Pringled committed Feb 20, 2025
1 parent 72105a5 commit cb94131
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model2vec/train/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ The scores are competitive with the popular [roberta-base-go_emotions](https://h

## Explainability

We offer a simple explainability method that allows you to see the most important tokens for a prediction. This is based on the logit outputs for the tokens in the input text, which we extract by forward passing them individually through the trained classifier. Since our classifier is a simple mean embedding followed by a single linear layer (meaning there is interaction between tokens), this is a good approximation of the importance of each token. The following code example shows how this works:
We offer a simple explainability method that allows you to see the most important tokens for a prediction. This is based on the logit outputs for the tokens in the input text, which we extract by forward passing them individually through the trained classifier. Since our classifier is a simple mean embedding followed by a single linear layer (meaning there is no interaction between tokens), this is a good approximation of the importance of each token. The following code example shows how this works:

```python
from datasets import load_dataset
Expand Down

0 comments on commit cb94131

Please sign in to comment.