msr2903's picture
Update README.md
1c1516e verified
metadata
library_name: transformers
datasets:
  - NickyNicky/finance-financialmodelingprep-stock-news-sentiments-rss-feed
base_model:
  - mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis

Model Description

This model is a fine-tuned version of mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis on the NickyNicky/finance-financialmodelingprep-stock-news-sentiments-rss-feed dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4090
  • Accuracy: 0.9171

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • num_epochs: 5

Training results

Training Loss Epoch Validation Loss
0.318500 1.0 0.294045
0.281700 2.0 0.298364
0.250100 3.0 0.302255
0.186400 4.0 0.380530
0.179100 5.0 0.409072