CoolFace
Modelpublic

LinkLinkWu/Boss_Stock_News_Analysis

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes17downloads
Model Card

๐Ÿง  FinancialBERT Sentiment Analysis (FinNews Binary)

This is a fine-tuned BERT model for binary sentiment classification of financial news headlines, adapted for real-time stock market sentiment prediction.

๐Ÿ” Model Details

  • โ€”Architecture: BERT (12-layer, 768-hidden, 12-heads)
  • โ€”Base model: ahmedrachid/FinancialBERT-Sentiment-Analysis
  • โ€”Fine-tuned task: Binary classification โ€“ Positive or Negative
  • โ€”Problem type: single_label_classification
  • โ€”Special tokens: [CLS], [SEP], [PAD], [MASK], [UNK]

Neutral headlines are mapped to Positive to simplify binary output.

๐Ÿงพ Training Summary

  • โ€”Dataset: 5,000+ manually labeled financial news headlines
  • โ€”Tokenizer: Custom WordPiece tokenizer
  • โ€”Max sequence length: 128
  • โ€”Framework: Transformers v4.51.3 (PyTorch backend)
  • โ€”Output labels:
  • โ€”LABEL_0 = Negative
  • โ€”LABEL_1 = Positive

๐Ÿ“ˆ Intended Use

Ideal for:

  • โ€”Real-time market sentiment dashboard
  • โ€”Trading signal pipelines
  • โ€”Event-driven NLP analysis

๐Ÿš€ Usage (Example)

python
from transformers import pipeline

classifier = pipeline("text-classification", model="your-username/your-model-name")
classifier("Apple's Q4 earnings beat expectations amid strong iPhone sales")
# Output: [{'label': 'LABEL_1', 'score': 0.98}]