Views
No views yet
1from transformers import BertTokenizer, BertForSequenceClassification, pipeline
2
3finbert = BertForSequenceClassification.from_pretrained('ZiweiChen/FinBERT-FOMC',num_labels=3)
4tokenizer = BertTokenizer.from_pretrained('ZiweiChen/FinBERT-FOMC')
5finbert_fomc = pipeline("text-classification", model=finbert, tokenizer=tokenizer)
6
7sentences = ["Spending on cars and light trucks increased somewhat in July after a lackluster pace in the second quarter but apparently weakened in August"]
8results = finbert_fomc(sentences)
9print(results)
10# [{'label': 'Negative', 'score': 0.994509756565094}]
11