知识图谱-第三方工具:OpenUE【轻量级知识图谱抽取工具(命名实体识别、关系抽取)】【基于PyTorchLightning+Transformers】

一、OpenUE主要架构

在这里插入图片描述

1、models 模块

其存放了我们主要的三个模型,针对整句的关系识别模型,针对已知句中关系的命名实体
识别模型,还有将前两者整合起来的推理验证模型。

2、lit_models 模块

其中的代码主要继承⾃pytorch_lightning.Trainer。其可以⾃动构建单卡,多卡,GPU,TPU
等不同硬件下的模型训练。

3、data 模块

data中存放了针对不同数据集进⾏不同操作的代码。使⽤了transformers库中的tokenizer先 对数据进⾏分词处理再根据不同需要将数据变成我们需要的features。

二、数据形式(json)

在这里插入图片描述


版权声明:本文为u013250861原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。