繁体   English   中英

如何在引导的LDA中生成术语矩阵以进行主题建模?

[英]How to generate term matrix in guided LDA for topic modeling?

我目前正在分析在线评论。 我想尝试GuidedLDA( https://medium.freecodecamp.org/how-we-changed-unsupervised-lda-to-semi-supervised-guidedlda-e36a95f3a164 ),因为一些主题重叠。 我已经成功安装了该软件包。 但是,我不确定如何使用excel文档作为输入来生成文档术语矩阵(在网站代码中称为X)和vocab。 有人可以帮忙吗? 我尝试在各种论坛上进行在线搜索,但未找到任何有效的方法。

从文本挖掘包中摘录,TDM类

汇入

导入csv

导入操作系统

'''

进口茎

'''

您可以将以下代码另存为单独的python文件,并将其作为常规模块导入您的代码中,例如create_tdm.py

导入create_tdm

X = create_tdm.TermDocumentMatrix(“您的文本”)

'''为Vocab'''

word2id = dict((id的(v,idx),enumerate(“您的文本”)中的v)

'''

确保在您的文本中包含引导词列表,否则您将得到关键错误,仅用于检查导入的熊猫是否为pd

c = pd.DataFrame(list(word2id))

'''

TermDocumentMatrix(object)类:

"""
Class to efficiently create a term-document matrix.

The only initialization parameter is a tokenizer function, which should
take in a single string representing a document and return a list of
strings representing the tokens in the document. If the tokenizer
parameter is omitted it defaults to using textmining.simple_tokenize

Use the add_doc method to add a document (document is a string). Use the
write_csv method to output the current term-document matrix to a csv
file. You can use the rows method to return the rows of the matrix if
you wish to access the individual elements without writing directly to a
file.

"""

def __init__(self, tokenizer=simple_tokenize):
    """Initialize with tokenizer to split documents into words."""
    # Set tokenizer to use for tokenizing new documents
    self.tokenize = tokenizer
    # The term document matrix is a sparse matrix represented as a
    # list of dictionaries. Each dictionary contains the word
    # counts for a document.
    self.sparse = []
    # Keep track of the number of documents containing the word.
    self.doc_count = {}

def add_doc(self, document):
    """Add document to the term-document matrix."""
    # Split document up into list of strings
    words = self.tokenize(document)
    # Count word frequencies in this document
    word_counts = {}
    for word in words:
        word_counts[word] = word_counts.get(word, 0) + 1
    # Add word counts as new row to sparse matrix
    self.sparse.append(word_counts)
    # Add to total document count for each word
    for word in word_counts:
        self.doc_count[word] = self.doc_count.get(word, 0) + 1

def rows(self, cutoff=2):
    """Helper function that returns rows of term-document matrix."""
    # Get master list of words that meet or exceed the cutoff frequency
    words = [word for word in self.doc_count \
      if self.doc_count[word] >= cutoff]
    # Return header
    yield words
    # Loop over rows
    for row in self.sparse:
        # Get word counts for all words in master list. If a word does
        # not appear in this document it gets a count of 0.
        data = [row.get(word, 0) for word in words]
        yield data

def write_csv(self, filename, cutoff=2):
    """
    Write term-document matrix to a CSV file.

    filename is the name of the output file (e.g. 'mymatrix.csv').
    cutoff is an integer that specifies only words which appear in
    'cutoff' or more documents should be written out as columns in
    the matrix.

    """
    f = csv.writer(open(filename, 'wb'))
    for row in self.rows(cutoff=cutoff):
        f.writerow(row)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM