lucas-leme

/

FinBERT-PT-BR

like 16

Também é possível utilizar o modelo lucas-leme/FinBERT-PT-BR com a função pipeline() . Segue um exemplo:

from transformers import (
    AutoTokenizer, 
    BertForSequenceClassification,
    pipeline,
)

finbert_pt_br_tokenizer = AutoTokenizer.from_pretrained("lucas-leme/FinBERT-PT-BR")
finbert_pt_br_model = BertForSequenceClassification.from_pretrained("lucas-leme/FinBERT-PT-BR")

finbert_pt_br_pipeline = pipeline(task='text-classification', model=finbert_pt_br_model, tokenizer=finbert_pt_br_tokenizer)
finbert_pt_br_pipeline(['Hoje a bolsa caiu', 'Hoje a bolsa subiu'])

Resultado

[{'label': 'NEGATIVE', 'score': 0.7304733395576477},
 {'label': 'POSITIVE', 'score': 0.5405582189559937}]