azalea/PySenti
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added dual and trinary

Browse Source
This commit is contained in:
zhunhung
2019-11-09 10:44:28 +08:00
parent 9dc59441b4
commit 4fcb106953
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sentistrength/__init__.py
+6 -7
View File
@@ -8,9 +8,6 @@ from os import getcwd
class PySentiStr: class PySentiStr:
def __init__(self): def __init__(self):
pass pass
# self.SentiStrengthLocation = os.path.join(getcwd(),"SentiStrength.jar")
# self.SentiStrengthLanguageFolder = os.path.join(getcwd(),"SentiStrengthData/")
def setSentiStrengthPath(self, ss_Path): def setSentiStrengthPath(self, ss_Path):
self.SentiStrengthLocation = ss_Path self.SentiStrengthLocation = ss_Path
@@ -40,19 +37,21 @@ class PySentiStr:
stdout_text = stdout_text.rstrip().replace("\t"," ") stdout_text = stdout_text.rstrip().replace("\t"," ")
stdout_text = stdout_text.replace('\r\n','') stdout_text = stdout_text.replace('\r\n','')
senti_score = stdout_text.split(' ') senti_score = stdout_text.split(' ')
try: try:
senti_score = list(map(float, senti_score)) senti_score = list(map(float, senti_score))
except ValueError: except ValueError:
raise ValueError("SentiStrengthLanguageFolderPath is set as '{}'. Ensure it is correct and ends with a forward slash '/'".format( self.SentiStrengthLanguageFolder)) raise ValueError("SentiStrengthLanguageFolderPath is set as '{}'. Ensure it is correct and ends with a forward slash '/'".format( self.SentiStrengthLanguageFolder))
senti_score = [int(i) for i in senti_score] senti_score = [int(i) for i in senti_score]
if score == 'scale': if score == 'scale': # Returns from -1 to 1
senti_score = [sum(senti_score[i:i+2])/4 for i in range(0, len(senti_score), 3)] senti_score = [sum(senti_score[i:i+2])/4 for i in range(0, len(senti_score), 3)]
elif score == 'binary': # Return Positive and Negative Score elif score == 'binary': # Return 1 if positive and -1 if negative
senti_score = [tuple(senti_score[i:i+2]) for i in range(0, len(senti_score), 3)] senti_score = [1 if senti_score[i] >= abs(senti_score[i+1]) else -1 for i in range(0, len(senti_score), 3)]
elif score == 'trinary': # Return Positive and Negative Score and Neutral Score elif score == 'trinary': # Return Positive and Negative Score and Neutral Score
senti_score = [tuple(senti_score[i:i+3]) for i in range(0, len(senti_score), 3)] senti_score = [tuple(senti_score[i:i+3]) for i in range(0, len(senti_score), 3)]
elif score == 'dual': # Return Positive and Negative Score
senti_score = [tuple(senti_score[i:i+2]) for i in range(0, len(senti_score), 3)]
else: else:
return "Argument 'score' takes in either 'scale' (between -1 to 1) or 'binary' (two scores, positive and negative rating)" return "Argument 'score' takes in either 'scale' (between -1 to 1) or 'binary' (two scores, positive and negative rating)"
return senti_score return senti_score