平均 Word2Vec 创建向量时出错

Posted

技术标签:

【中文标题】平均 Word2Vec 创建向量时出错【英文标题】:Getting error on averaging Word2Vec crerated vectors 【发布时间】:2020-07-22 23:34:09 【问题描述】:

我想使用 gensim 在我的推文数据集上创建 Word2Vec 向量。该代码用于基于推文的多标签情感分类。我汇总了包含 107k 条推文的推文文件。我用它来创建基于的 Word2Vec 向量。我的代码:

np.set_printoptions(threshold=sys.maxsize)

#Pre-Processor Function
pre_processor = TextPreProcessor(
    omit=['url', 'email', 'percent', 'money', 'phone', 'user',
        'time', 'url', 'date', 'number'],
    
    normalize=['url', 'email', 'percent', 'money', 'phone', 'user',
        'time', 'url', 'date', 'number'],
     
    segmenter="twitter", 
    
    corrector="twitter", 
    
    unpack_hashtags=True,
    unpack_contractions=True,
    
    tokenizer=SocialTokenizer(lowercase=True).tokenize,
    
    dicts=[emoticons]
)

#Averaging Words Vectors to Create Sentence Embedding
def word_averaging(wv, words):
    all_words, mean = set(), []
    
    for word in words:
        if isinstance(word, np.ndarray):
            mean.append(word)
        elif word in wv.vocab:
            mean.append(wv.syn0norm[wv.vocab[word].index])
            all_words.add(wv.vocab[word].index)

    if not mean:
        logging.warning("cannot compute similarity with no input %s", words)
        # FIXME: remove these examples in pre-processing
        return np.zeros(wv.vector_size,)

    mean = gensim.matutils.unitvec(np.array(mean).mean(axis=0)).astype(np.float32)
    return mean

def  word_averaging_list(wv, text_list):
    return np.vstack([word_averaging(wv, post) for post in text_list ])

#Loading data
raw_aggregate_tweets = pandas.read_excel('E:\\aggregate.xlsx').iloc[:,0] #Loading all tweets to have a bigger word2vec corpus

raw_train_tweets = pandas.read_excel('E:\\train.xlsx').iloc[:,1] #Loading all train tweets
train_labels = np.array(pandas.read_excel('E:\\train.xlsx').iloc[:,2:13]) #Loading corresponding train labels (11 emotions)

raw_test_tweets = pandas.read_excel('E:\\test.xlsx').iloc[:,1] #Loading all test tweets
test_gold_labels = np.array(pandas.read_excel('E:\\test.xlsx').iloc[:,2:13]) #Loading corresponding test labels (11 emotions)
print("please wait")

#Pre-Processing
aggregate_tweets=[]
train_tweets=[]
test_tweets=[]
for tweets in raw_aggregate_tweets:
    aggregate_tweets.append(pre_processor.pre_process_doc(tweets))

for tweets in raw_train_tweets:
    train_tweets.append(pre_processor.pre_process_doc(tweets))

for tweets in raw_test_tweets:
    test_tweets.append(pre_processor.pre_process_doc(tweets))


#Vectorizing 
w2v_model = gensim.models.Word2Vec(aggregate_tweets, min_count = 10, size = 300, window = 8)


train_array = word_averaging_list(w2v_model.wv,train_tweets)
test_array = word_averaging_list(w2v_model.wv,test_tweets)

但我收到此错误:

TypeError                                 Traceback (most recent call last)
<ipython-input-1-8a5fe4dbf144> in <module>
    110 print(w2v_model.wv.vectors.shape)
    111 
--> 112 train_array = word_averaging_list(w2v_model.wv,train_tweets)
    113 test_array = word_averaging_list(w2v_model.wv,test_tweets)
    114 

<ipython-input-1-8a5fe4dbf144> in word_averaging_list(wv, text_list)
     70 
     71 def  word_averaging_list(wv, text_list):
---> 72     return np.vstack([word_averaging(wv, post) for post in text_list ])
     73 
     74 #Averaging Words Vectors to Create Sentence Embedding

<ipython-input-1-8a5fe4dbf144> in <listcomp>(.0)
     70 
     71 def  word_averaging_list(wv, text_list):
---> 72     return np.vstack([word_averaging(wv, post) for post in text_list ])
     73 
     74 #Averaging Words Vectors to Create Sentence Embedding

<ipython-input-1-8a5fe4dbf144> in word_averaging(wv, words)
     58             mean.append(word)
     59         elif word in wv.vocab:
---> 60             mean.append(wv.syn0norm[wv.vocab[word].index])
     61             all_words.add(wv.vocab[word].index)
     62 

TypeError: 'NoneType' object is not subscriptable

看起来您的帖子主要是代码;请添加更多细节。这个网站的错误是什么?天哪。我没有更多细节。抱歉,我必须这样做才能绕过错误。

二次平均法

#Averaging Words Vectors to Create Sentence Embedding
def get_mean_vector(word2vec_model, words):
    # remove out-of-vocabulary words
    words = [word for word in words if word in word2vec_model.vocab]
    if len(words) >= 1:
        return np.mean(word2vec_model[words], axis=0)
    else:
        return np.zeros(word2vec_model.vector_size)

#Vectorizing 
w2v_model = gensim.models.Word2Vec(aggregate_tweets, min_count = 11, size = 400, window = 18, sg=1)

train_array=[]
test_array=[]
for tweet in train_tweets:
    vec = get_mean_vector(w2v_model.wv, tweet)
    if len(vec) > 0:
        train_array.append(vec)
        
for tweet in test_tweets:
    vec = get_mean_vector(w2v_model.wv, tweet)
    if len(vec) > 0:
        test_array.append(vec)

【问题讨论】:

您的问题到底是什么?你做过调试吗?第二种方法与第一种方法有什么关系? 【参考方案1】:

错误“'NoneType' object is not subscriptable”表示您尝试下标(使用[] 进行索引访问)实际上是None 的变量。

查看突出显示的行,wv.syn0norm 可能是None

它不会自动存在:它只在需要时创建,例如通过.most_similar() 操作。但是你可以手动触发它的创建,一旦你的训练完成,通过调用.init_sims()

w2v_model.wv.init_sims()

(请注意,您可能会从代码中收到弃用警告:在最近的 gensim 版本中,该属性已重命名为 vectors_norm。此外,出于某些目的,使用这些单位长度归一化向量可能不太好,作为原始向量。)

【讨论】:

太棒了!有效。但是由于我是这个领域的新手并且不太了解您的提示,所以我尝试了另一种更简单的平均方法的代码。请看一下第二个平均方法部分并告诉我您对此的看法。与前一种方法相比,第二种方法使我的准确率提高了 1%。谢谢。 是的,这是第二个平均函数。 还有必要再次使用w2v_model.wv.init_sims()吗? 你真的应该尝试一下,看看有没有错误。它应该比询问更快、更容易! 你的解释我不太明白。这就是为什么我再次问。现在我明白你的意思了。再次感谢。

以上是关于平均 Word2Vec 创建向量时出错的主要内容,如果未能解决你的问题,请参考以下文章

如何使用 word2vec 找到最接近向量的单词

word2vec概述

怎样用word2vec来得到某几个词的向量表示

word2vec改进之Hierarchical Softmax

如何使用 Word2Vec 获取单词列表的向量?

基于word2vec的文档向量模型的应用