YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

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
1
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support