专栏首页机器学习AI算法工程开源中文关系抽取框架,来自浙大知识引擎实验室

开源中文关系抽取框架,来自浙大知识引擎实验室

DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。

环境依赖:

python >= 3.6

  • torch >= 1.2
  • hydra-core >= 0.11
  • tensorboard >= 2.0
  • matplotlib >= 3.1
  • transformers >= 2.0
  • jieba >= 0.39

中文关系抽取

  1. 基于 CNN 的关系抽取模型
  2. 基于 BiLSTM 的关系抽取模型
  3. 基于 PCNN 的远程监督关系抽取模型
  4. 基于 Capsule 的关系抽取模型
  5. 基于 Transformer 的关系抽取模型
  6. 基于 GCN 的关系抽取模型
  7. 基于 BERT 语言预训练模型的关系抽取模型

AI项目体验地址 https://loveai.tech

快速开始

数据为 csv 文件,样式范例为:

  • 安装依赖: pip install -r requirements.txt
  • 存放数据:在 data/origin 文件夹下存放训练数据。训练文件主要有三个文件。更多数据建议使用百度数据库中Knowledge Extraction。
    • train.csv:存放训练数据集
    • valid.csv:存放验证数据集
    • test.csv:存放测试数据集
    • relation.csv:存放关系种类
  • 开始训练:python main.py
  • 每次训练的日志保存在 logs 文件夹内,模型结果保存在 checkpoints 文件夹内

本文分享自微信公众号 - 机器学习AI算法工程(datayx)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2020-09-22

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 短实体,长句实体抽取

    “半指针-半标注”方法实体的抽取器,基于苏神的三元组抽取方法改造,这里取消了三元组抽取模型中对s的抽取,直接抽取实体并做分类(相当于直接抽取p和o)。改造后的实...

    机器学习AI算法工程
  • 知识图谱入门 , 知识抽取

    知识抽取,即从不同来源、不同结构的数据中进行知识提取,形成知识(结构化数据)存入到知识图谱。大体的任务分类与对应技术如下图所示:

    机器学习AI算法工程
  • 基于深度学习的自动化行人检测和监控系统

    展示系统基于Idea集成开发环境进行开发,SSM框架中的依赖均基于Maven进行配置,在Idea中导入web目录下的工程,导出war包,将war包放在服务器to...

    机器学习AI算法工程
  • Unity Package包内插件解锁

    新版的Unity将模块工具与游戏中的资源文件分开放置,但有一个问题,里边的插件都是只读的,无法添加内容,连创建都是灰色的orz:

    汐夜koshio
  • Linux下使用 github+hexo 搭建个人博客02-hexo部署到Github Pages

    之前的这篇文章《Linux下使用 github+hexo 搭建个人博客01-hexo搭建》,相信大家都知道怎么搭建 hexo ,怎么切换主题,并且完成了一篇博文...

    踏歌行
  • 云计算的限制:为什么需要边缘的机器学习

    能够在设备或边缘上执行基于机器学习的任务,而不是将其发送到云端进行处理,许多人将其描述为“机器学习演进的下一个阶段”。有许多重要的限制,功耗、成本、延迟、隐私等...

    静一
  • 知识图谱从哪里来:实体关系抽取的现状与未来

    最近几年深度学习引发的人工智能浪潮席卷全球,在互联网普及带来的海量数据资源和摩尔定律支配下飞速提升的算力资源双重加持下,深度学习深入影响了自然语言处理的各个方向...

    Datawhale
  • Spark内部原理

    Spark中的Shuffle、宽依赖窄依赖、RDD持久化、共享变量

    俺也想起舞
  • 夏季达沃斯热议第四次工业革命,二手车大佬成黑马

    6月26日到6月28日,世界经济论坛第十届新领军者年会(夏季达沃斯)在天津举办,来自90个国家的1700多位政府和企业领袖围绕第四次工业革命对各国经济、各行各业...

    罗超频道
  • 知识图谱从哪里来:实体关系抽取的现状与未来

    最近几年深度学习引发的人工智能浪潮席卷全球,在互联网普及带来的海量数据资源和摩尔定律支配下飞速提升的算力资源双重加持下,深度学习深入影响了自然语言处理的各个方向...

    AI科技大本营

扫码关注云+社区

领取腾讯云代金券