知识图谱-第三方工具：OpenUE【轻量级知识图谱抽取工具（命名实体识别、关系抽取）】【基于PyTorchLightning+Transformers】
发布时间
阅读量：
阅读量
一、OpenUE主要架构
1、models 模块
该系统存储了我们主要的三个模型：其一用于对整句话的关系识别任务；其二是专门处理句子内关系命名实体识别的任务；此外还集成了一个综合推理验证系统。
2、lit_models 模块
其中的代码主要基于pytorch_lightning.Trainer框架设计。该代码能够支持单卡、多卡、GPU、TPU等多种硬件环境下的模型训练。
3、data 模块
data中存储了对应不同数据集实施多样化的操作代码。采用了transformers库中的tokenizer首先对数据进行分词处理随后根据具体需求将这些分词后的数据转换成所需的features。
二、数据形式(json)
全部评论 (0)
还没有任何评论哟~