开源中文关系抽取框架，来自浙大知识引擎实验室
共 1950字，需浏览 4分钟
·
2020-09-22 20:33
向AI转型的程序员都关注了这个号???
机器学习AI算法工程 公众号：datayx
DeepKE
DeepKE 是基于 Pytorch 的深度学习中文关系抽取处理套件。
环境依赖:
python >= 3.6
torch >= 1.2
hydra-core >= 0.11
tensorboard >= 2.0
matplotlib >= 3.1
transformers >= 2.0
jieba >= 0.39
中文关系抽取
基于 CNN 的关系抽取模型
基于 BiLSTM 的关系抽取模型
基于 PCNN 的远程监督关系抽取模型
基于 Capsule 的关系抽取模型
基于 Transformer 的关系抽取模型
基于 GCN 的关系抽取模型
基于 BERT 语言预训练模型的关系抽取模型
代码 以及运行教程 获取：
关注微信公众号 datayx 然后回复 抽取 即可获取。
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文件夹内
阅读过本文的人还看了以下文章：
基于40万表格数据集TableBank，用MaskRCNN做表格检测
《深度学习入门：基于Python的理论与实现》高清中文PDF+源码
2019最新《PyTorch自然语言处理》英、中文版PDF+源码
《21个项目玩转深度学习：基于TensorFlow的实践详解》完整版PDF+附书代码
PyTorch深度学习快速实战入门《pytorch-handbook》
【下载】豆瓣评分8.1,《机器学习实战:基于Scikit-Learn和TensorFlow》
李沐大神开源《动手学深度学习》，加州伯克利深度学习（2019春）教材
【Keras】完整实现‘交通标志’分类、‘票据’分类两个项目，让你掌握深度学习图像分类
如何利用全新的决策树集成级联结构gcForest做特征工程并打分？
Machine Learning Yearning 中文翻译稿
斯坦福CS230官方指南：CNN、RNN及使用技巧速查（打印收藏）
中科院Kaggle全球文本匹配竞赛华人第1名团队-深度学习与特征工程
不断更新资源
深度学习、机器学习、数据分析、python
搜索公众号添加： datayx
机大数据技术与机器学习工程
搜索公众号添加： datanlp
长按图片，识别二维码