YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
Fine-Tuned Transformer Model
This is a fine-tuned Transformer model for predicting stock trends based on labeled news headlines. It predicts the following classes:
- Downward (0)
- Sideways (1)
- Upward (2)
Training Details
- Learning Rate: 2e-5
- Batch Size: 8
- Epochs: 15
Metrics
- Accuracy
- Precision
- Recall
- F1 Score
- Downloads last month
- 8