如何从gensim文件中删除停用词?

时间:2017-06-22 12:02:20

标签: python nlp gensim word2vec doc2vec

我正在使用其gensim包在Python中使用Doc2Vec技术构建一个NLP聊天应用程序。我已经完成了令牌化和词干化。我想从训练集中删除停用词(以测试它是否更好),以及用户抛出的问题。

这是我的代码。

import gensim
import nltk
from gensim import models
from gensim import utils
from gensim import corpora
from nltk.stem import PorterStemmer
ps = PorterStemmer()

sentence0 = models.doc2vec.LabeledSentence(words=[u'sampl',u'what',u'is'],tags=["SENT_0"])
sentence1 = models.doc2vec.LabeledSentence(words=[u'sampl',u'tell',u'me',u'about'],tags=["SENT_1"])
sentence2 = models.doc2vec.LabeledSentence(words=[u'elig',u'what',u'is',u'my'],tags=["SENT_2"])
sentence3 = models.doc2vec.LabeledSentence(words=[u'limit', u'what',u'is',u'my'],tags=["SENT_3"])
sentence4 = models.doc2vec.LabeledSentence(words=[u'claim',u'how',u'much',u'can',u'I'],tags=["SENT_4"])
sentence5 = models.doc2vec.LabeledSentence(words=[u'retir',u'i',u'am',u'how',u'much',u'can',u'elig',u'claim'],tags=["SENT_5"])
sentence6 = models.doc2vec.LabeledSentence(words=[u'resign',u'i',u'have',u'how',u'much',u'can',u'i',u'claim',u'elig'],tags=["SENT_6"])
sentence7 = models.doc2vec.LabeledSentence(words=[u'promot',u'what',u'is',u'my',u'elig',u'post',u'my'],tags=["SENT_7"])
sentence8 = models.doc2vec.LabeledSentence(words=[u'claim',u'can,',u'i',u'for'],tags=["SENT_8"])
sentence9 = models.doc2vec.LabeledSentence(words=[u'product',u'coverag',u'cover',u'what',u'all',u'are'],tags=["SENT_9"])
sentence10 = models.doc2vec.LabeledSentence(words=[u'hotel',u'coverag',u'cover',u'what',u'all',u'are'],tags=["SENT_10"])
sentence11 = models.doc2vec.LabeledSentence(words=[u'onlin',u'product',u'can',u'i',u'for',u'bought',u'through',u'claim',u'sampl'],tags=["SENT_11"])
sentence12 = models.doc2vec.LabeledSentence(words=[u'reimburs',u'guidelin',u'where',u'do',u'i',u'apply',u'form',u'sampl'],tags=["SENT_12"])
sentence13 = models.doc2vec.LabeledSentence(words=[u'reimburs',u'procedur',u'rule',u'and',u'regul',u'what',u'is',u'the',u'for'],tags=["SENT_13"])
sentence14 = models.doc2vec.LabeledSentence(words=[u'can',u'i',u'submit',u'expenditur',u'on',u'behalf',u'of',u'my',u'friend',u'and',u'famili',u'claim',u'and',u'reimburs'],tags=["SENT_14"])
sentence15 = models.doc2vec.LabeledSentence(words=[u'invoic',u'bills',u'procedur',u'can',u'i',u'submit',u'from',u'shopper stop',u'claim'],tags=["SENT_15"])
sentence16 = models.doc2vec.LabeledSentence(words=[u'invoic',u'bills',u'can',u'i',u'submit',u'from',u'pantaloon',u'claim'],tags=["SENT_16"])
sentence17 = models.doc2vec.LabeledSentence(words=[u'invoic',u'procedur',u'can',u'i',u'submit',u'invoic',u'from',u'spencer',u'claim'],tags=["SENT_17"])

# User asks a question.

document = input("Ask a question:")
tokenized_document = list(gensim.utils.tokenize(document, lowercase = True, deacc = True))
#print(type(tokenized_document))
stemmed_document = []
for w in tokenized_document:
    stemmed_document.append(ps.stem(w))
sentence19 = models.doc2vec.LabeledSentence(words= stemmed_document, tags=["SENT_19"])

# Building vocab.
sentences = [sentence0,sentence1,sentence2,sentence3, sentence4, sentence5,sentence6, sentence7, sentence8, sentence9, sentence10, sentence11, sentence12, sentence13, sentence14, sentence15, sentence16, sentence17, sentence19]

#I tried to remove the stop words but it didn't work out as LabeledSentence object has no attribute lower.
stoplist = set('for a of the and to in'.split())
texts = [[word for word in document.lower().split() if word not in stoplist]
          for document in sentences]
..

有没有办法可以直接从sentences删除停用词并获得一套新的词汇而不用停字?

1 个答案:

答案 0 :(得分:3)

您的sentences对象已经是LabeledSentence个对象的列表。你构建了以上这些;它们包括words中的字符串列表和tags中的字符串列表。

因此,该列表中的每个项目(列表解析中的document)都不能应用.lower()这样的字符串方法。 (也不需要.split(),因为words已经是单独的令牌。)

最干净的方法是在用于构建LabeledSentence对象之前从列表中删除停用词。例如,您可以创建一个在顶部定义的函数without_stopwords()。然后创建LabeledSentence对象的行可以改为:

sentence0 = LabeledSentence(words=remove_stopwords([u'sampl', u'what', u'is']), 
                            tags=["SENT_0"])

或者,你可以改变现有的LabeledSentence个对象,这样它们的每个words属性现在都没有停用词。这将取代你的最后一行更像是:

for doc in sentences:
    doc.words = [word for word in doc.words if word not in stoplist]
texts = sentences

另外,你没有提出但应该知道的事情:

  • TaggedDocument现在是Doc2Vec文本对象的首选示例类名称 - 但实际上任何具有两个必需属性wordstags的对象都可以正常工作。

  • Doc2Vec并没有在微小的,玩具大小的数据集上显示许多所需的属性 - 如果基于几十个句子的模型没有做任何有用的事情,或者误导有关,那么不要感到惊讶什么预处理/元参数选项是最好的。 (成千上万的文本和至少几十个字的文本,对于有意义的结果来说要好得多。)

  • 很多Word2Vec / Doc2Vec的工作并不需要去除词干或停止词,但它有时可能会有所帮助。