File size: 507 Bytes
be5fbad
 
 
 
 
625c998
 
8581867
625c998
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
---
language:
- eu
---

# Basque Semantic Textual Similarity Model

```python
import transformers

if __name__ == '__main__':
    model = 'eu-sts'
    tokenizer = transformers.AutoTokenizer.from_pretrained(model)
    model = transformers.BertForSequenceClassification.from_pretrained(model)

    sentence_pairs = [("Nire ama etxera etorri da.", "Ama visitan etorri da etxera")]
    for sp in sentence_pairs:
        result = model(**tokenizer(*sp, return_tensors='pt'))
        print(sp, result.logits)

```