jonas/nlp-lab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed missing return

Browse Source
This commit is contained in:
Jonas Weinz 2018-06-26 20:33:55 +02:00
parent b8bd92db06
commit 480c1e8928
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Project/simple_approach/simple_twitter_learning.py
View File

@ -136,6 +136,7 @@ def lemm(s):
word_lemmatized = lemmatizer.lemmatize(word[0], pos=wordnet_pos)
lemmatized_sent.append(word_lemmatized)
lemmatized_sent = (" ").join(lemmatized_sent)
return lemmatized_sent
def batch_stem(sentences):
@ -575,9 +576,9 @@ class pipeline_manager(object):
def predict(self,X, use_stemming=True, use_lemmatization=True):
"""predict"""
if use_stemming:
X = batch_stem(X)
X = np.array(batch_stem(X))
if use_lemmatization:
X = batch_lemm(X)
X = np.array(batch_lemm(X))
return self.pipeline.predict(X)