pytorch_casrel_triple_extraction

KO | EN

gitlite — search

#javascript #python #react #typescript #llm #android #deep-learning #linux #docker #java #windows #pytorch #security #ios

pytorch_casrel_triple_extraction

(★ 43)

基于pytorch的CasRel进行三元组抽取。

pytorch_casrel_triple_extraction 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# pytorch_casrel_triple_extraction
基于pytorch的CasRel进行三元组抽取。

之前的[基于pytorch的中文三元组提取（命名实体识别+关系抽取）](https://github.com/taishan1994/pytorch_triple_extraction)是先抽取主体和客体，再进行关系分类，这里是使用casrel，先抽取主体，再识别客体和关系。具体使用说明：

- 1、在data/ske/raw_data下是原始数据，新建一个process.py，主要是得到mid_data下的关系的类型。
- 2、针对于不同的数据源，在data_loader.py中修改MyDataset类下，返回的是一个列表，列表中的每个元素是：(text, labels)，其中labels是[[主体，类别，客体]]。
- 3、运行main.py进行训练、验证、测试和预测。

数据和模型下载地址：链接：https://pan.baidu.com/s/12w8vC1Arfo8FTH3AuN1WAw?pwd=h7hu  提取码：h7hu

# 依赖

```
pytorch==1.6.0
transformers==4.5.0
```

# 运行

```python
python main.py \
--bert_dir="model_hub/chinese-bert-wwm-ext/" \
--data_dir="./data/ske/" \
--log_dir="./logs/" \
--output_dir="./checkpoints/" \
--num_tags=49 \  # 关系类别
--seed=123 \
--gpu_ids="0" \
--max_seq_len=256 \  # 句子最大长度
--lr=5e-5 \
--other_lr=5e-5 \
--train_batch_size=32 \
--train_epochs=1 \
--eval_batch_size=8 \
--max_grad_norm=1 \
--warmup_proportion=0.1 \
--adam_epsilon=1e-8 \
--weight_decay=0.01 \
--dropout_prob=0.1 \
--use_tensorboard="False" \  # 是否使用tensorboardX可视化
--use_dev_num=1000 \ # 使用多少验证数据进行验证
```

### 结果
这里使用batch_size=32训练了3000步。
```
========metric========
precision:0.7087378640776699 recall:0.7237960339943342 f1:0.7161878065872459
```
```python

文本： 查尔斯·阿兰基斯（Charles Aránguiz），1989年4月17日出生于智利圣地亚哥，智利职业足球运动员，司职中场，效力于德国足球甲级联赛勒沃库森足球俱乐部
主体： [['查尔斯·阿兰基斯']]
客体： [['智利', '智利圣地亚哥', '1989年4月17日']]
关系： [[('查尔斯·阿兰基斯', '国籍', '智利'), ('查尔斯·阿兰基斯', '出生地', '智利圣地亚哥'), ('查尔斯·阿兰基斯', '出生日期', '1989年4月17日')]]
====================================================================================================
文本： 《离开》是由张宇谱曲，演唱
主体： [['离开']]
客体： [['张宇']]
关系： [[('离开', '作词', '张宇'), ('离开', '歌手', '张宇'), ('离开', '作曲', '张宇')]]
====================================================================================================
文本： 《愤怒的唐僧》由北京吴意波影视文化工作室与优酷电视剧频道联合制作，故事以喜剧元素为主，讲述唐僧与佛祖打牌，得罪了佛祖，被踢下人间再渡九九八十一难的故事
主体： [['愤怒的唐僧']]
客体： [['北京吴意波影视文化工作室']]
关系： [[('愤怒的唐僧', '出品公司', '北京吴意波影视文化工作室')]]
====================================================================================================
文本： 李治即位后，萧淑妃受宠，王皇后为了排挤萧淑妃，答应李治让身在感业寺的武则天续起头发，重新纳入后宫
主体： [['萧淑妃']]
客体： [[]]
关系： [[]]
====================================================================================================
文本： 《工业4.0》是2015年机械工业出版社出版的图书，作者是（德）阿尔冯斯·波特霍夫，恩斯特·安德雷亚斯·哈特曼
主体： [['工业4.0']]
客体： [['阿尔冯斯·波特霍夫', '恩斯特·安德雷亚斯·哈特曼', '机械工业出版社']]
关系： [[('工业4.0', '作者', '阿尔冯斯·波特霍夫'), ('工业4.0', '作者', '恩斯特·安德雷亚斯·哈特曼'), ('工业4.0', '出版社', '机械工业出版社')]]
====================================================================================================
文本： 周佛海被捕入狱之后，其妻杨淑慧散尽家产请蒋介石枪下留人，于是周佛海从死刑变为无期，不过此人或许作恶多端，改判没多久便病逝于监狱，据悉是心脏病发作
主体： [['周佛海', '杨淑慧']]
客体： [[]]
关系： [[]]
====================================================================================================
文本： 《李烈钧自述》是2011年11月1日人民日报出版社出版的图书，作者是李烈钧
主体： [['李烈钧自述']]
客体： [['人民日报出版社']]
关系： [[('李烈钧自述', '出版社', '人民日报出版社')]]
====================================================================================================
文本： 除演艺事业外，李冰冰热心公益，发起并亲自参与多项环保慈善活动，积极投身其中，身体力行担起了回馈社会的责任于02年出演《少年包青天》，进入大家视线
主体： [['少年包青天']]
客体： [['李冰冰']]
关系： [[('少年包青天', '主演', '李冰冰')]]
====================================================================================================
文本： 马志舟，1907年出生，陕西三原人，汉族，中国共产党，任红四团第一连连长，1933年逝世
主体： [['马志舟']]
客体： [['中国', '陕西三原', '1907年', '汉族']]
关系： [[('马志舟', '国籍', '中国'), ('马志舟', '出生地', '陕西三原'), ('马志舟', '出生日期', '1907年'), ('马志舟', '民族', '汉族')]]
====================================================================================================
文本： 斑刺莺是雀形目、剌嘴莺科的一种动物，分布于澳大利亚和新西兰，包括澳大利亚、新西兰、塔斯马尼亚及其附近的岛屿
主体： [['斑刺莺']]
客体： [['雀形目']]
关系： [[('斑刺莺', '目', '雀形目')]]
====================================================================================================
文本： 《课本上学不到的生物学2》是2013年上海科技教育出版社出版的图书
主体： [['课本上学不到的生物学2']]
客体： [['上海科技教育出版社']]
关系： [[('课本上学不到的生物学2', '出版社', '上海科技教育出版社')]]
====================================================================================================
```

# 参考

> 模型参考：[bert4torch/task_relation_extraction.py at master · Tongjilibo/bert4torch (github.com)](https://github.com/Tongjilibo/bert4torch/blob/master/examples/relation_extraction/task_relation_extraction.py)

pytorch_casrel_triple_extraction

pytorch_casrel_triple_extraction 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

pytorch_casrel_triple_extraction

pytorch_casrel_triple_extraction 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README