Refactor sentiment calculation in reward.py

nanlabs · Feb 26, 2024 · 9f655fc · 9f655fc
1 parent 46ca4fd
commit 9f655fc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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(
-Original file line number
+Diff line change
@@ Expand Up / @@ -56,7 +56,6 @@ async def forward(self): @@
             - Rewarding the miners
             - Updating the scores
             """
             return await forward(self)
@@ Expand Down @@