Views
No views yet
pipeline1from transformers import pipeline
2
3classifier = pipeline(
4 "text-classification",
5 model="seu-usuario/Sentiments-FinBERT-PT-BR"
6)
7
8resultado = classifier("As ações da empresa subiram após o balanço positivo.")
9print(resultado)AutoModel1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("seu-usuario/Sentiments-FinBERT-PT-BR")
5model = AutoModelForSequenceClassification.from_pretrained("seu-usuario/Sentiments-FinBERT-PT-BR")
6
7texto = "O lucro líquido da companhia caiu 30% no trimestre."
8inputs = tokenizer(texto, return_tensors="pt", truncation=True, padding=True)
9
10with torch.no_grad():
11 logits = model(**inputs).logits
12
13predicao = torch.argmax(logits, dim=-1).item()
14print(model.config.id2label[predicao])SANTOS, Lucas L.; BIANCHI, Reinaldo A. C.; COSTA, Anna H. R. FinBERT-PT-BR: Análise de Sentimentos de Textos em Português do Mercado Financeiro. Anais do II Brazilian Workshop on Artificial Intelligence in Finance, 2023, p. 144–155.
1@inproceedings{santos2023finbertptbr,
2 title={FinBERT-PT-BR: An{\'a}lise de Sentimentos de Textos em Portugu{\^e}s do Mercado Financeiro},
3 author={Santos, Lucas L. and Bianchi, Reinaldo A. C. and Costa, Anna H. R.},
4 booktitle={Anais do II Brazilian Workshop on Artificial Intelligence in Finance},
5 pages={144--155},
6 year={2023}
7}