|
--- |
|
language: es |
|
tags: |
|
- zero-shot-classification |
|
- nli |
|
- pytorch |
|
datasets: |
|
- xnli |
|
pipeline_tag: zero-shot-classification |
|
license: apache-2 |
|
widget: |
|
- text: "El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo" |
|
candidate_labels: "cultura, sociedad, economia, salud, deportes" |
|
--- |
|
# ZERO-SHOT SELECTRA: A zero-shot classifier based on SELECTRA |
|
|
|
## Usage |
|
|
|
```python |
|
from transformers import pipeline |
|
classifier = pipeline("zero-shot-classification", |
|
model="Recognai/zeroshot_selectra_medium") |
|
|
|
classifier( |
|
"El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo", |
|
candidate_labels=["cultura", "sociedad", "economia", "salud", "deportes"], |
|
hypothesis_template="Este ejemplo es {}." |
|
) |
|
``` |
|
|
|
## Training |
|
|
|
For the training notebook, please check out our [repository](https://github.com/recognai/selectra/tree/main/zero-shot_classifier). |