Merge pull request #65 from zengliangju/patch-2

Update __init__.py
isnowfy · May 19, 2017 · 8c2cc92 · 8c2cc92
2 parents efc19ba + 9bf2a7e
commit 8c2cc92
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/snownlp/sentiment/__init__.py b/snownlp/sentiment/__init__.py
@@ -48,8 +48,14 @@ def classify(self, sent):
 
 
 def train(neg_file, pos_file):
-    neg_docs = codecs.open(neg_file, 'r', 'utf-8').readlines()
-    pos_docs = codecs.open(pos_file, 'r', 'utf-8').readlines()
+    neg = codecs.open(neg_file, 'r', 'utf-8').readlines()
+    pos = codecs.open(pos_file, 'r', 'utf-8').readlines()
+    neg_docs = []
+    pos_docs = []
+    for line in neg:
+        neg_docs.append(line.rstrip("\r\n"))
+    for line in pos:
+        pos_docs.append(line.rstrip("\r\n"))
     global classifier
     classifier = Sentiment()
     classifier.train(neg_docs, pos_docs)