gensim中word2vec API参数说明

最新推荐文章于 2024-06-28 19:39:46 发布

清萝卜头

最新推荐文章于 2024-06-28 19:39:46 发布

阅读量5.1k

点赞数 6

CC 4.0 BY-SA版权

分类专栏： NLP 文章标签： Word2Vec gensim 参数说明

本文链接：https://blog.csdn.net/xiaoQL520/article/details/102509477

这篇博客详细介绍了gensim库中word2vec模型的相关API，重点在于gensim.models.word2vec.Word2Vec类的参数说明。内容包括算法参数的翻译及解释，帮助理解如何在gensim中应用word2vec进行词向量训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在gensim中，word2vec相关的API都在gensim.models.word2vec中，与算法相关的参数在 gensim.models.word2vec.Word2Vec中。其具体参数说明如下：（个人翻译，如有出入欢迎指正）

class Word2Vec(BaseWordEmbeddingsModel):
    """训练, 使用和评估https://code.google.com/p/word2vec/中描述的神经网络

    一旦你训练完一个模型 (不再更新,仅查询)
    只能在`~gensim.models.keyedvectors.KeyedVectors`的`self.wv`实例中存储并使用，这样可以节省内存

    该模型可以通过`~gensim.models.word2vec.Word2Vec.save`和`~gensim.models.word2vec.Word2Vec.load`方法来存储和加载

    训练后的单词向量还可以通过`self.wv.save_word2vec_format`原始word2vec实现方法和`gensim.models.keyedvectors.KeyedVectors.load_word2vec_format`来存储和加载。

    一些重要的属性如下：
    属性
    ----------
    wv : :class:`~gensim.models.keyedvectors.Word2VecKeyedVectors`
        这个对象本质上包含单词和embeeding之间的映射。经过训练，可以通过各种方式直接用来查询这些embeeding。有关示例，请参见模块级文档说明。

    vocabulary : :class:`~gensim.models.word2vec.Word2VecVocab`
        该对象表示模型的词汇表（有时在gensim中称为Dictionary）。
        除了跟踪所有不同的单词之外，此对象还