Views
No views yet
SentenceTransformer(
(0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: XLMRobertaModel
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
)1import torch
2from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
4tok = AutoTokenizer.from_pretrained("mawaskow/inc_sent_cls_bn")
5model = AutoModelForSequenceClassification.from_pretrained("mawaskow/inc_sent_cls_bn")
6
7sentences = [
8 "The authority can revise the delegated act every five years.",
9 "The scheme will subsidise purchases of eco-friendly farm equipment.",
10 "Farmers will be able to avail of expert assistance in the uptake of new technologies."
11]
12text = sentences[1]
13inputs = tok(text, return_tensors="pt")
14
15with torch.no_grad():
16 logits = model(**inputs).logits
17
18pred = torch.argmax(logits, dim=-1).item()
19print(model.config.id2label[pred])
20# incentive