Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer, AutoConfig
2import numpy as np
3import torch
4
5model_name = "sagui-nlp/debertinha-ptbr-xsmall-assin2-sts"
6s1 = "A gente faz o aporte financeiro, é como se a empresa fosse parceira do Monte Cristo."
7s2 = "Fernando Moraes afirma que não tem vínculo com o Monte Cristo além da parceira."
8model = AutoModelForSequenceClassification.from_pretrained(model_name)
9tokenizer = AutoTokenizer.from_pretrained(model_name)
10config = AutoConfig.from_pretrained(model_name)
11model_input = tokenizer(*([s1], [s2]), padding=True, return_tensors="pt")
12with torch.no_grad():
13 output = model(**model_input)
14 score = output[0][0].detach().numpy().item()
15 print(f"Similarity Score: {np.round(float(score), 4)}")@misc{campiotti2023debertinha,
title={DeBERTinha: A Multistep Approach to Adapt DebertaV3 XSmall for Brazilian Portuguese Natural Language Processing Task},
author={Israel Campiotti and Matheus Rodrigues and Yuri Albuquerque and Rafael Azevedo and Alyson Andrade},
year={2023},
eprint={2309.16844},
archivePrefix={arXiv},
primaryClass={cs.CL}
}