ckharche commited on
Commit
eed5e46
·
verified ·
1 Parent(s): 0713532

Update trade_analysis/enhanced_sentiment.py

Browse files
Files changed (1) hide show
  1. trade_analysis/enhanced_sentiment.py +12 -12
trade_analysis/enhanced_sentiment.py CHANGED
@@ -38,24 +38,24 @@ class EnhancedFinancialSentimentAnalyzer:
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': {
 
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': {