|
--- |
|
language: en |
|
--- |
|
|
|
# CTRL44 Classification model |
|
|
|
This is a pretrained version of the 4-class simplification operation classifier presented in the NAACL 2022 paper "Controllable Sentence Simplification via Operation Classification". It was trained on the IRSD classification dataset. |
|
|
|
Predictions from this model can be used for input into the [simplification model](https://huggingface.co/liamcripwell/ctrl44-simp) to reproduce pipeline results seen in the paper. |
|
|
|
## How to use |
|
|
|
Here is how to use this model in PyTorch: |
|
|
|
```python |
|
from transformers import RobertaForSequenceClassification, AutoTokenizer |
|
|
|
model = RobertaForSequenceClassification.from_pretrained("liamcripwell/ctrl44-clf") |
|
tokenizer = AutoTokenizer.from_pretrained("liamcripwell/ctrl44-clf") |
|
|
|
text = "Barack Hussein Obama II is an American politician who served as the 44th president of the United States from 2009 to 2017." |
|
inputs = tokenizer(text, return_tensors="pt") |
|
|
|
with torch.no_grad(): |
|
logits = model(**inputs).logits |
|
predicted_class_id = logits.argmax().item() |
|
predicted_class_name = model.config.id2label[predicted_class_id] |
|
``` |
