harshithakr commited on
Commit
c140769
·
1 Parent(s): ca37c6b

Update preprocess_function.py

Browse files
Files changed (1) hide show
  1. preprocess_function.py +3 -2
preprocess_function.py CHANGED
@@ -82,5 +82,6 @@ def bert_preprocess(keyword):
82
  def lam_list(list_words):
83
  list_words = [x.strip().lower() for x in list_words]
84
  lemmatizer = WordNetLemmatizer()
85
- list_words = [lemmatizer.lemmatize(word,pos='v') for word in list_words]
86
- return list_words
 
 
82
  def lam_list(list_words):
83
  list_words = [x.strip().lower() for x in list_words]
84
  lemmatizer = WordNetLemmatizer()
85
+ list_words_v = [lemmatizer.lemmatize(word,pos='v') for word in list_words]
86
+ list_words_n = [lemmatizer.lemmatize(word,pos='n') for word in list_words]
87
+ return list_words_v, list_words_n