一、OpenUE主要架构
![在这里插入图片描述](https://img-blog.csdnimg.cn/a89c77c448bf44cd93c19431910ead65.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5b-N6ICF44Gu5Lmx5aSq6YOO,size_20,color_FFFFFF,t_70,g_se,x_16)
1、models 模块
其存放了我们主要的三个模型,针对整句的关系识别模型,针对已知句中关系的命名实体
识别模型,还有将前两者整合起来的推理验证模型。
2、lit_models 模块
其中的代码主要继承⾃pytorch_lightning.Trainer。其可以⾃动构建单卡,多卡,GPU,TPU
等不同硬件下的模型训练。
3、data 模块
data中存放了针对不同数据集进⾏不同操作的代码。使⽤了transformers库中的tokenizer先 对数据进⾏分词处理再根据不同需要将数据变成我们需要的features。
二、数据形式(json)
![在这里插入图片描述](https://img-blog.csdnimg.cn/6c388da9c9974a26a32ef0af13afbde0.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5b-N6ICF44Gu5Lmx5aSq6YOO,size_20,color_FFFFFF,t_70,g_se,x_16)