Views
No views yet
INTRO, BACK, METH, RES, DISC, CONC, CONTR, LIM.Flaglab/SciBETO-largedocumento_id, seed=42| Métrica | 4 clases | 8 clases |
|---|---|---|
| F1-macro | 0.5781 | 0.7083 |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("wiflore/SciBETO-IMRaD")
5model = AutoModelForSequenceClassification.from_pretrained("wiflore/SciBETO-IMRaD")
6
7text = "En este estudio proponemos un método para..."
8inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
9with torch.no_grad():
10 logits = model(**inputs).logits
11pred = logits.argmax(-1).item()
12labels = ['INTRO','BACK','METH','RES','DISC','CONC','CONTR','LIM']
13print(labels[pred])