Views
No views yet

1from transformers import AutoTokenizer, BertForSequenceClassification
2import numpy as np
3
4pred_mapper = {
5 0: "POSITIVE",
6 1: "NEGATIVE",
7 2: "NEUTRAL"
8 }
9
10tokenizer = AutoTokenizer.from_pretrained("lucas-leme/FinBERT-PT-BR")
11finbertptbr = BertForSequenceClassification.from_pretrained("lucas-leme/FinBERT-PT-BR")
12
13tokens = tokenizer(["Hoje a bolsa caiu", "Hoje a bolsa subiu"], return_tensors="pt",
14 padding=True, truncation=True, max_length=512)
15finbertptbr_outputs = finbertptbr(**tokens)
16
17preds = [pred_mapper[np.argmax(pred)] for pred in finbertptbr_outputs.logits.cpu().detach().numpy()]1from transformers import (
2 AutoTokenizer,
3 BertForSequenceClassification,
4 pipeline,
5)
6
7finbert_pt_br_tokenizer = AutoTokenizer.from_pretrained("lucas-leme/FinBERT-PT-BR")
8finbert_pt_br_model = BertForSequenceClassification.from_pretrained("lucas-leme/FinBERT-PT-BR")
9
10finbert_pt_br_pipeline = pipeline(task='text-classification', model=finbert_pt_br_model, tokenizer=finbert_pt_br_tokenizer)
11finbert_pt_br_pipeline(['Hoje a bolsa caiu', 'Hoje a bolsa subiu'])1@inproceedings{santos2023finbert,
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 AC and Costa, Anna HR},
4 booktitle={Anais do II Brazilian Workshop on Artificial Intelligence in Finance},
5 pages={144--155},
6 year={2023},
7 organization={SBC}
8}