takala/financial_phrasebank
Updated • 8.66k • 259
How to use d4niel92/phrasebank-sentiment-analysis with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="d4niel92/phrasebank-sentiment-analysis") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("d4niel92/phrasebank-sentiment-analysis")
model = AutoModelForSequenceClassification.from_pretrained("d4niel92/phrasebank-sentiment-analysis")This model is a fine-tuned version of bert-base-uncased on the financial_phrasebank dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | F1 | Accuracy |
|---|---|---|---|---|---|
| 0.5844 | 0.94 | 100 | 0.4858 | 0.7733 | 0.8122 |
| 0.2654 | 1.89 | 200 | 0.3783 | 0.8456 | 0.8514 |
| 0.1362 | 2.83 | 300 | 0.4873 | 0.8532 | 0.8583 |
| 0.0653 | 3.77 | 400 | 0.5829 | 0.8411 | 0.8459 |
Base model
google-bert/bert-base-uncased