我从一些python任务开始,我在使用gensim时遇到了问题。我正在尝试从我的磁盘加载文件并处理它们(拆分它们和小写()它们)
我的代码如下:
dictionary_arr=[]
for file_path in glob.glob(os.path.join(path, '*.txt')):
with open (file_path, "r") as myfile:
text=myfile.read()
for words in text.lower().split():
dictionary_arr.append(words)
dictionary = corpora.Dictionary(dictionary_arr)
列表(dictionary_arr)包含所有文件中所有单词的列表,然后我使用gensim corpora.Dictionary来处理列表。但是我遇到了错误。
TypeError: doc2bow expects an array of unicode tokens on input, not a single string
我无法理解什么是问题,我们将不胜感激。
答案 0 :(得分:8)
在dictionary.py中,initialize函数是:
def __init__(self, documents=None):
self.token2id = {} # token -> tokenId
self.id2token = {} # reverse mapping for token2id; only formed on request, to save memory
self.dfs = {} # document frequencies: tokenId -> in how many documents this token appeared
self.num_docs = 0 # number of documents processed
self.num_pos = 0 # total number of corpus positions
self.num_nnz = 0 # total number of non-zeroes in the BOW matrix
if documents is not None:
self.add_documents(documents)
函数add_documents从文档集合构建字典。每个文档都是一个列表 代币:
def add_documents(self, documents):
for docno, document in enumerate(documents):
if docno % 10000 == 0:
logger.info("adding document #%i to %s" % (docno, self))
_ = self.doc2bow(document, allow_update=True) # ignore the result, here we only care about updating token ids
logger.info("built %s from %i documents (total %i corpus positions)" %
(self, self.num_docs, self.num_pos))
因此,如果以这种方式初始化Dictionary,则必须传递文档但不能传递单个文档。例如,
dic = corpora.Dictionary([a.split()])
没问题。
答案 1 :(得分:3)
大家好,我遇到了同样的问题。这就是对我有用的
#Tokenize the sentence into words
tokens = [word for word in sentence.split()]
#Create dictionary
dictionary = corpora.Dictionary([tokens])
print(dictionary)
答案 2 :(得分:1)
Dictionary的输入需要使用标记字符串:
dataset = ['driving car ',
'drive car carefully',
'student and university']
# be sure to split sentence before feed into Dictionary
dataset = [d.split() for d in dataset]
vocab = Dictionary(dataset)