Downloads · 30 days
7
1% of all-time downloads
vidhivaish03/Sentiment_Analysis_Stock_Tweets_FineTuned
Sentiment_Analysis_Stock_Tweets_FineTuned is a text classification model from vidhivaish03. Use it when you need a label for a piece of text. It is set up for transformers.
Model Name: vidhivaish03/SentimentAnalysisStockTweetsFineTuned Model Type: Transformer-based, fine-tuned Base Model: ahmedrachid/FinancialBERT-Sentiment-Analysis Language: English Task: Sentiment Analysis (Bearish, Bu…
Downloads · 30 days
7
1% of all-time downloads
All-time downloads
501
Public
Parameters
110M
439 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors439 MB · 100%
From the Hugging Face model README
Model Name: vidhivaish03/Sentiment_Analysis_Stock_Tweets_FineTuned
Model Type: Transformer-based, fine-tuned
Base Model: ahmedrachid/FinancialBERT-Sentiment-Analysis
Language: English
Task: Sentiment Analysis (Bearish, Bullish, Neutral)
Dataset Name: zeroshot/twitter-financial-news-sentiment
Description: The dataset consists of financial news tweets annotated with three sentiment labels: Bearish, Bullish, and Neutral. It includes 9,543 training samples and 2,388 validation samples.
Data Structure:
Preprocessing: Text cleaning (removal of URLs, special characters), tokenization, stop word removal, lemmatization, and vectorization using Word2Vec embeddings.
Training Framework: Hugging Face's Transformers library
Fine-Tuning Epochs: 5
Optimizer: AdamW
Learning Rate: 2e-5
The classification report provides a detailed performance overview of the fine-tuned FinancialBERT model on the validation set, evaluating its ability to classify stock-related tweets into Bearish, Bullish, and Neutral sentiments.
Precision Recall F1-Score Support
Bearish 0.83 0.69 0.76 285
Bullish 0.75 0.83 0.79 391
Neutral 0.91 0.91 0.91 1233
accuracy 0.86 1909
macro avg 0.83 0.81 0.82 1909
weighted avg 0.86 0.86 0.86 1909
Hugging Face Hub: The model is available on Hugging Face Hub for easy integration and deployment.
Inference: The model can be accessed via RESTful APIs or integrated into financial sentiment analysis pipelines for real-time predictions.
Repository Link: vidhivaish03/Sentiment_Analysis_Stock_Tweets_FineTuned
This model card summarizes the key aspects of vidhivaish03/Sentiment_Analysis_Stock_Tweets_FineTuned, providing insights into its training, performance, and applications.