Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

change remove_numbers_fun with change_numbers_fun #1

Open
wants to merge 5 commits into
base: master
Choose a base branch
from
Open

change remove_numbers_fun with change_numbers_fun #1

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
274056d
change remove_numbers_fun with change_numbers_fun
MubashirullahD Mar 27, 2020
cf887d7
improve regex for html tag removal
MubashirullahD Mar 28, 2020
3106ecd
Merge pull request #1 from MubashirullahD/patch-2
MubashirullahD Mar 28, 2020
d8a1e71
Revert "improve regex for html tag removal"
MubashirullahD Mar 28, 2020
449b1fa
Merge pull request #2 from MubashirullahD/revert-1-patch-2
MubashirullahD Mar 28, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions nlppreprocess/nlppreprocess.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@

class NLP():
def __init__(self, remove_stopwords=True, replace_words=True,
remove_numbers=True, remove_html_tags=True,
replace_numbers=True, remove_html_tags=True,
remove_punctuations=True, lemmatize=False,
lemmatize_method='wordnet'):
"""
Expand All @@ -36,7 +36,7 @@ def __init__(self, remove_stopwords=True, replace_words=True,
"""
if (type(remove_stopwords) != bool or
type(replace_words) != bool or
type(remove_numbers) != bool or
type(replace_numbers) != bool or
type(remove_html_tags) != bool or
type(remove_punctuations) != bool or
type(lemmatize) != bool):
Expand All @@ -47,7 +47,7 @@ def __init__(self, remove_stopwords=True, replace_words=True,
self.lemmatizer = None
self.remove_stopwords = remove_stopwords
self.replace_words = replace_words
self.remove_numbers = remove_numbers
self.replace_numbers = replace_numbers
self.remove_html_tags = remove_html_tags
self.remove_punctations = remove_punctuations
self.lemmatize_method = lemmatize_method
Expand Down Expand Up @@ -87,12 +87,12 @@ def replace_words_fun(self):
cleaned_doc.append(word)
self.doc = ' '.join(cleaned_doc)

def remove_numbers_fun(self):
def replace_numbers_fun(self):
"""
This function uses regex to remve
all the numbers from the doc.
"""
self.doc = re.sub("[0-9]", "", self.doc)
self.doc = re.sub("[0-9]+", "number", self.doc)

def remove_html_tags_fun(self):
"""
Expand Down Expand Up @@ -230,8 +230,8 @@ def process(self, doc):
self.remove_html_tags_fun()
if self.remove_stopwords is True:
self.remove_stopwords_fun()
if self.remove_numbers is True:
self.remove_numbers_fun()
if self.replace_numbers is True:
self.replace_numbers_fun()
if self.remove_punctations is True:
self.remove_punctations_fun()
if self.lemmatize is True:
Expand Down