Spaces:
Sleeping
Sleeping
Update trade_analysis/enhanced_sentiment.py
Browse files
trade_analysis/enhanced_sentiment.py
CHANGED
|
@@ -38,24 +38,24 @@ class EnhancedFinancialSentimentAnalyzer:
|
|
| 38 |
# 'type': 'classification',
|
| 39 |
# 'specialization': 'general_financial'
|
| 40 |
# },
|
| 41 |
-
'finbert_tone': {
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
},
|
| 47 |
# 'roberta_financial': {
|
| 48 |
# 'model_id': 'cardiffnlp/twitter-roberta-base-sentiment-latest',
|
| 49 |
# 'weight': 0.20,
|
| 50 |
# 'type': 'classification',
|
| 51 |
# 'specialization': 'social_sentiment'
|
| 52 |
# },
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
|
| 60 |
# # Tier 2: Specialized Models
|
| 61 |
# 'fintwit_bert': {
|
|
|
|
| 38 |
# 'type': 'classification',
|
| 39 |
# 'specialization': 'general_financial'
|
| 40 |
# },
|
| 41 |
+
# 'finbert_tone': {
|
| 42 |
+
# 'model_id': 'yiyanghkust/finbert-tone',
|
| 43 |
+
# 'weight': 0.25,
|
| 44 |
+
# 'type': 'classification',
|
| 45 |
+
# 'specialization': 'tone_analysis'
|
| 46 |
+
# },
|
| 47 |
# 'roberta_financial': {
|
| 48 |
# 'model_id': 'cardiffnlp/twitter-roberta-base-sentiment-latest',
|
| 49 |
# 'weight': 0.20,
|
| 50 |
# 'type': 'classification',
|
| 51 |
# 'specialization': 'social_sentiment'
|
| 52 |
# },
|
| 53 |
+
'distilroberta_financial': {
|
| 54 |
+
'model_id': 'mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis',
|
| 55 |
+
'weight': 0.20,
|
| 56 |
+
'type': 'classification',
|
| 57 |
+
'specialization': 'news_sentiment'
|
| 58 |
+
},
|
| 59 |
|
| 60 |
# # Tier 2: Specialized Models
|
| 61 |
# 'fintwit_bert': {
|