从gensim word2Vec获取权重矩阵

我在Python中使用gensim的word2vec包。我希望获取在skip-gram学习过程中学习到的WW'权重矩阵。

我认为model.syn0可以给我第一个矩阵,但我不确定如何获取第二个矩阵。有什么想法吗?

实际上,我非常希望找到关于模型可访问属性的详尽文档,因为官方文档似乎不够精确(例如,syn0没有被描述为一个属性)


回答:

model.wv.syn0包含了输入嵌入矩阵。当使用层次化softmaxhs=1)训练时,输出嵌入存储在model.syn1中;当使用负采样(negative>0)时,存储在model.syn1neg中。就是这样!当层次化softmax和负采样都没有启用时,Word2Vec使用单一的权重矩阵model.wv.syn0进行训练。

另见相关讨论这里

Related Posts

Keras Dense层输入未被展平

这是我的测试代码: from keras import…

无法将分类变量输入随机森林

我有10个分类变量和3个数值变量。我在分割后直接将它们…

如何在Keras中对每个输出应用Sigmoid函数?

这是我代码的一部分。 model = Sequenti…

如何选择类概率的最佳阈值?

我的神经网络输出是一个用于多标签分类的预测类概率表: …

在Keras中使用深度学习得到不同的结果

我按照一个教程使用Keras中的深度神经网络进行文本分…

‘MatMul’操作的输入’b’类型为float32,与参数’a’的类型float64不匹配

我写了一个简单的TensorFlow代码,但不断遇到T…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注