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

Fixed Name Error

Browse Source
This commit is contained in:
zhunhung
2019-04-05 12:46:55 +08:00
parent cd6d6362cf
commit 59c0a5ce02
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sentistrength/__init__.py
+9 -7
View File
@@ -7,14 +7,11 @@ from os import getcwd
class PySentiStr:
def __init__(self):
self.SentiStrengthLocation = os.path.join(getcwd(),"SentiStrength.jar")
self.SentiStrengthLanguageFolder = os.path.join(getcwd(),"SentiStrengthData/")
if not os.path.isfile(self.SentiStrengthLocation):
print("SentiStrength not found at: ", SentiStrengthLocation,'\nSet path using setSentiStrengthPath(path) function.')
if not os.path.isdir(self.SentiStrengthLanguageFolder):
print("SentiStrength data folder not found at: ", SentiStrengthLanguageFolder,'\nSet path using setSentiStrengthLanguageFolderPath(path) function.')
pass
# self.SentiStrengthLocation = os.path.join(getcwd(),"SentiStrength.jar")
# self.SentiStrengthLanguageFolder = os.path.join(getcwd(),"SentiStrengthData/")
def setSentiStrengthPath(self, ss_Path):
self.SentiStrengthLocation = ss_Path
@@ -22,6 +19,11 @@ class PySentiStr:
self.SentiStrengthLanguageFolder = sslf_Path
def getSentiment(self, df_text, score='scale'):
if not hasattr(self, 'SentiStrengthLocation'):
print("Set path using setSentiStrengthPath(path) function.")
if not hasattr(self, 'SentiStrengthLanguageFolder'):
print("Set path using setSentiStrengthLanguageFolderPath(path) function.")
# Able to take in single string or list of string and convert into pandas Series
if type(df_text) != pd.Series:
df_text = pd.Series(df_text)