From 9f655fc22d29aabf75a959d1d420ae40307f697f Mon Sep 17 00:00:00 2001 From: ulises-jeremias Date: Mon, 26 Feb 2024 14:39:29 -0300 Subject: [PATCH] Refactor sentiment calculation in reward.py --- neurons/validators/validator.py | 1 - prompting/validator/reward.py | 4 +++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/neurons/validators/validator.py b/neurons/validators/validator.py index 07f722a..4537908 100644 --- a/neurons/validators/validator.py +++ b/neurons/validators/validator.py @@ -56,7 +56,6 @@ async def forward(self): - Rewarding the miners - Updating the scores """ - return await forward(self) diff --git a/prompting/validator/reward.py b/prompting/validator/reward.py index 1b852af..91d6f1b 100644 --- a/prompting/validator/reward.py +++ b/prompting/validator/reward.py @@ -34,7 +34,9 @@ def reward(query: int, response: str) -> float: blob = textblob.TextBlob(response) sentiment_sum = reduce(lambda x, y: x + y, [sentence.sentiment.polarity for sentence in blob.sentences]) - return sentiment_sum / len(blob.sentences) + sentiment_avg = sentiment_sum / len(blob.sentences) + sentiment_normalized = (sentiment_avg + 1) / 2 + return sentiment_normalized def get_rewards(