Create lemmatizing.py

Author: pemagrg1

lemmatizing.py

@@ -0,0 +1,24 @@
+# coding: utf-8
+import nltk
+from nltk.corpus import stopwords
+from nltk.stem import WordNetLemmatizer
+
+stopword = stopwords.words('english')
+wordnet_lemmatizer = WordNetLemmatizer()
+
+def lemmatize(text):
+    """
+    take string input and lemmatize the words.
+    use WordNetLemmatizer to lemmatize the words.
+    """
+    word_tokens = nltk.word_tokenize(text)
+    lemmatized_word = [wordnet_lemmatizer.lemmatize(word) for word in word_tokens]
+    return (lemmatized_word)
+
+def main():
+    text = "the functions of this fan is awesome"
+    print (remove_punct(text))
+
+if __name__ == '__main__':
+    main()
+