从维基百科学习单词和实体的矢量表示的工具

wikipedia2vec-SM002583的Python项目详细描述


维基百科2vec

Fury badgeCircleCI

wikipedia2vec是一个工具,用于从wikipedia获取单词和实体(即wikipedia中具有相应页面的概念)的嵌入(或向量表示)。 它由Studio Ousia开发和维护。

此工具使您能够同时学习单词和实体的嵌入,并将相似的单词和实体彼此靠近放置在连续向量空间中。 嵌入可以很容易地通过一个命令进行训练,并以一个公开可用的wikipedia转储作为输入。

这个工具实现了conventional skip-gram model来学习单词的嵌入,以及Yamada et al. (2016)中提出的扩展来学习实体的嵌入。 此工具已用于几个最新的NLP模型,如entity linkingnamed entity recognitionknowledge graph completionentity relatednessquestion answering

这个工具已经在linux、windows和macos上测试过了。

< > WiKiTaa2VEC与现有的嵌入工具(即FAST文本、GENSIM、RDF2VEC和WiKi2VEC)的经验比较是可用的{{A11}。

12种语言(英语、阿拉伯语、中文、荷兰语、法语、德语、意大利语、日语、波兰语、葡萄牙语、俄语和西班牙语)的文档和预训练嵌入可在http://wikipedia2vec.github.io/在线获得。

基本用法

维基百科2vec可以通过pypi安装:

% pip install wikipedia2vec

有了这个工具,可以通过运行一个train命令并以wikipedia dump作为输入来学习嵌入。 例如,以下命令下载最新的英语维基百科转储并从此转储学习嵌入:

% wget https://dumps.wikimedia.org/enwiki/latest/enwiki-latest-pages-articles.xml.bz2
% wikipedia2vec train enwiki-latest-pages-articles.xml.bz2 MODEL_FILE

然后,学习的嵌入被写入model_文件。 请注意,此命令可以接受许多可选参数。 详情请参阅our documentation

参考

如果您在科学出版物中使用Wikipedia2vec,请引用以下文章:

山田英彦,浅井昭一,新藤弘一,竹田英彦,竹藤喜彦,Wikipedia2Vec: An Optimized Tool for Learning Embeddings of Words and Entities from Wikipedia

@article{yamada2018wikipedia2vec,
  title={Wikipedia2Vec: An Optimized Tool for Learning Embeddings of Words and Entities from Wikipedia},
  author={Yamada, Ikuya and Asai, Akari and Shindo, Hiroyuki and Takeda, Hideaki and Takefuji, Yoshiyasu},
  journal={arXiv preprint 1812.06280},
  year={2018}
}

许可证

Apache License 2.0

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
空字符串检查在java中未按预期工作   JavaSpringWebClient:自动计算主体的HMAC签名并将其作为头传递   foreach是否有一个Java等效的foreach循环和一个引用变量?   java如何在Eclipse中导入jar   使用特定第三方或java时lombok触发错误。*方法或构造函数   安卓 java将对象数组转换为int数组   java使一定百分比的JUnit测试通过   java Android:将Seekbar的一个值与另一个值进行比较   java将int数组(图像数据)写入文件的最佳方式是什么   java取代了系统。yml的构造函数内的getProperty   sqlite Java将公钥和私钥转换为字符串,然后再转换回字符串   安卓获取白色像素并将其保存到java opencv中的数组中   java为什么是ServerSocket。setSocketFactory静态?   Java数组似乎在不直接修改的情况下更改值