Question

在Gensim 2.2.0中使用具有IMDB影片评级的普通英文文本文件创建单词向量：

import gensim, logging
import smart_open, os
from nltk.tokenize import RegexpTokenizer

VEC_SIZE = 300 
MIN_COUNT = 5
WORKERS = 4
data_path = './data/'
vectors_path = 'vectors.bin.gz'

class AllSentences(object):
    def __init__(self, dirname):
        self.dirname = dirname
        self.read_err_cnt = 0
        self.tokenizer = RegexpTokenizer('[\'a-zA-Z]+', discard_empty=True)

    def __iter__(self):
        for fname in os.listdir(self.dirname):
            print(fname)
            for line in open(os.path.join(self.dirname, fname)):
                words = []     
                try:
                    for word in self.tokenizer.tokenize(line):
                        words.append(word)
                    yield words
                except:
                    self.read_err_cnt += 1

sentences = AllSentences(data_path)

培训和储蓄模式：

model = gensim.models.Word2Vec(sentences, size=VEC_SIZE, 
                               min_count=MIN_COUNT, workers=WORKERS)
word_vectors = model.wv
word_vectors.save(vectors_path)

然后尝试加载它：

vectors = KeyedVectors.load_word2vec_format(vectors_path,
                                                    binary=True,
                                                    unicode_errors='ignore')

我得到＆＃39; UnicodeDecode错误：＆＃39; utf-8＆＃39;编解码器不能解码位置0的字节0x80 ＆＃39;例外（见下文）。我尝试了不同的编码组合＆＃39;参数包括 ＆＃39; ISO-8859-1＆＃39; 和 ＆＃39; Latin1＆＃39; 。 ＆＃39; binary = True / False＆＃39; 的不同组合。没有什么可以帮助 - 无论使用什么参数，都是相同的例外。怎么了？如何使加载向量起作用？

例外：

UnicodeDecodeError                        Traceback (most recent call last)
<ipython-input-64-f353fa49685c> in <module>()
----> 1 w2v = get_w2v_vectors()

<ipython-input-63-cbbe0a76e837> in get_w2v_vectors()
      3     vectors = KeyedVectors.load_word2vec_format(word_vectors_path,
      4                                                     binary=True,
----> 5                                                     unicode_errors='ignore')
      6 
      7                                                 #unicode_errors='ignore')

D:\usr\anaconda\lib\site-packages\gensim\models\keyedvectors.py in load_word2vec_format(cls, fname, fvocab, binary, encoding, unicode_errors, limit, datatype)
    204         logger.info("loading projection weights from %s", fname)
    205         with utils.smart_open(fname) as fin:
--> 206             header = utils.to_unicode(fin.readline(), encoding=encoding)
    207             vocab_size, vector_size = map(int, header.split())  # throws for invalid file format
    208             if limit:

D:\usr\anaconda\lib\site-packages\gensim\utils.py in any2unicode(text, encoding, errors)
    233     if isinstance(text, unicode):
    234         return text
--> 235     return unicode(text, encoding, errors=errors)
    236 to_unicode = any2unicode
    237 

UnicodeDecodeError: 'utf-8' codec can't decode byte 0x80 in position 0: invalid start byte

Answer 1

如果使用gensim的原生save()方法保存向量，则应使用本机load()方法加载它们。

如果您想使用load_word2vec_format()加载向量，则需要使用save_word2vec_format()保存它们。（您将以这种方式丢失一些信息，例如确切的出现计数，否则将在KeyedVectors.vocab字典项目中。）

Gensim：单词向量编码问题

1 个答案: