Andreagus commited on
Commit
7221dc5
1 Parent(s): 3933b4d

Update aggregator.py

Browse files
Files changed (1) hide show
  1. aggregator.py +10 -2
aggregator.py CHANGED
@@ -76,7 +76,9 @@ def get_articles_sentiment(ticker, model):
76
  dict["label"] = 2
77
  else:
78
  dict["label"] = 1
79
-
 
 
80
  try:
81
  replace_values(bezinga_results)
82
 
@@ -205,6 +207,9 @@ def get_articles_sentiment(ticker, model):
205
  if len(vantage_negatives) > 0:
206
  vantage_negative_score_mean = float(sum(d['score'] for d in vantage_negatives)) / len(vantage_negatives)
207
 
 
 
 
208
  results_dict = {
209
  "bezinga": {
210
  "bezinga_articles": len(bezinga_results) if bezinga_results else 0,
@@ -253,7 +258,10 @@ def get_articles_sentiment(ticker, model):
253
  "vantage_sentiment_mean": vantage_label_mean if vantage_results else 0,
254
  "vantage_positive_score_mean": vantage_positive_score_mean if vantage_results else 0,
255
  "vantage_negative_score_mean": vantage_negative_score_mean if vantage_results else 0
256
- }
 
 
 
257
  }
258
 
259
  return results_dict
 
76
  dict["label"] = 2
77
  else:
78
  dict["label"] = 1
79
+
80
+ total_articles = len(bezinga_results) + len(finhub_results) + len(marketaux_results) + len(newsapi_results) + len(newsdata_results) + len(vantage_results)
81
+
82
  try:
83
  replace_values(bezinga_results)
84
 
 
207
  if len(vantage_negatives) > 0:
208
  vantage_negative_score_mean = float(sum(d['score'] for d in vantage_negatives)) / len(vantage_negatives)
209
 
210
+ total_positives = len(bezinga_positives) + len(finhub_positives) + len(marketaux_positives) + len(newsapi_positives) + len(newsdata_positives) + len(vantage_positives)
211
+ total_negatives = len(bezinga_negatives) + len(finhub_negatives) + len(marketaux_negatives) + len(newsapi_negatives) + len(newsdata_negatives) + len(vantage_negatives)
212
+
213
  results_dict = {
214
  "bezinga": {
215
  "bezinga_articles": len(bezinga_results) if bezinga_results else 0,
 
258
  "vantage_sentiment_mean": vantage_label_mean if vantage_results else 0,
259
  "vantage_positive_score_mean": vantage_positive_score_mean if vantage_results else 0,
260
  "vantage_negative_score_mean": vantage_negative_score_mean if vantage_results else 0
261
+ },
262
+ "total_articles": total_articles,
263
+ "total_positives": total_positives,
264
+ "total_negatives": total_negatives
265
  }
266
 
267
  return results_dict