替换 Word2Vec Gensim 中的目标函数

Replace objetive function in Word2Vec Gensim

我正在做我的最终学位项目。我需要创建 word2vec 算法的扩展版本,更改原始论文的默认 objective 函数。这已经完成了(检查这个 paper)。在那篇论文中,他们只说了新的 objective 功能,但他们没有说他们如何拥有 运行 模型。

现在,我也需要用另一个函数扩展那个模型,但我不确定我是否必须用新函数自己实现 word2vec,或者有一种方法可以在 Gensim word2vec 实现中替换它.

我已经检查了 Word2Vec Gensim documentation 但我没有看到任何参数来执行此操作。你知道怎么做吗?甚至有可能吗?

我不确定这个 StackExchange 站点是否正确,也许 https://ai.stackexchange.com/ 更合适。

Gensim 中没有官方支持直接放入您自己的 objective 函数。

但是,完整的源代码是可用的 – https://github.com/RaRe-Technologies/gensim – 所以通过编辑它,或者将它用作您自己实现的模型,理论上您可以做任何事情。

不过要注意:

  • 代码已经针对可能与您的需求不相关的新选项进行了大量优化和定制,因此可能不是最简洁的起点
  • 为了性能,核心例程是用 Cython 编写的——请参阅 .pyx 文件——这可能特别难以调试,并且依赖库批量数组函数,这些函数可能会掩盖如何实现你的替代函数