Update aggregator.py
Browse files- 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
|