'Word2Vec'对象没有属性'index2word'

时间:2018-10-03 14:14:25

标签: python-3.x text-mining word2vec

我在以下python代码中收到此错误“ AttributeError:'Word2Vec'对象没有属性'index2word'”。谁知道我该怎么解决? 正确地,“ tfidf_weighted_averaged_word_vectorizer”引发了错误。 “ obli.csv”包含句子行。 谢谢。

from feature_extractors import tfidf_weighted_averaged_word_vectorizer

    dataset = get_data2()
    corpus, labels = dataset.data, dataset.target
    corpus, labels = remove_empty_docs(corpus, labels)
    # print('Actual class label:', dataset.target_names[labels[10]])

    train_corpus, test_corpus, train_labels, test_labels = prepare_datasets(corpus,
                                                                            labels,
                                                                            test_data_proportion=0.3)
    tfidf_vectorizer, tfidf_train_features = tfidf_extractor(train_corpus)


    vocab = tfidf_vectorizer.vocabulary_
        tfidf_wv_train_features = tfidf_weighted_averaged_word_vectorizer(corpus=tokenized_train,
                                                                          tfidf_vectors=tfidf_train_features,
                                                                          tfidf_vocabulary=vocab,
                                                                          model=model,
                                                                          num_features=100)



    def get_data2():

        obli = pd.read_csv('db/obli.csv').values.ravel().tolist()
        cl0 = [0 for x in range(len(obli))]

        nonObli = pd.read_csv('db/nonObli.csv').values.ravel().tolist()
        cl1 = [1 for x in range(len(nonObli))]

        all = obli + nonObli


        db =  Db(all,cl0 + cl1)
        db.data = all
        db.target = cl0 + cl1

        return db

1 个答案:

答案 0 :(得分:1)

这是Dipanjan Sarkar撰写的Python文本分析的第4章中的代码。

自该文本发布以来,gensim中的

index2word已被移动。

您应该使用model.index2word而不是model.wv.index2word