2016-11-18 49 views
0

我一直在试图运行吨如何从蟒蛇的gensim库使用word2vec一个例子,但我不断收到此错误使用a.all()或a.any()错误,而试图使用gensim word2vec

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all() 

这是我的代码,它只是一个简单的例子:

from gensim.models import Word2Vec 
    sentences = [['first', 'sentence'], ['second', 'sentence']] 
    # train word2vec on the two sentences 
    model = Word2Vec(sentences, min_count=1) 

注:我确信,gensim安装其所有的依赖关系。

回答

0

我有同样的确切的问题太多,我所做的就是安装Python-dev软件包,然后重新安装gensim,反正那工作,我对Ubuntu的,所以这是我所做的:

sudo apt-get install python-dev 
sudo pip uninstall gensim 
sudo pip install gensim 

当我运行此:

model = gensim.models.Word2Vec(sentences=listSentence,min_count=2,window=3,size=20,workers=1) 
print model['Brasil'] 

它的工作和我得到的结果向量:

[-0.01635483 0.02224622 -0.01865266 0.02168317 -0.-0.0207897 
-0.0014509 0.00264822 -0.01889374 -0.02109174 -0.00244757 0.00024959 
-0.00898884 -0.01826199 -0.01361686 -0.01770178 -0.02431025 -0.01903439 
-0.00775641 0.02353667] 
+0

感谢您的回答,我有点想通OU也是。问题出自python-dev安装。 – Amani

相关问题