# DuEE-transformers

**Repository Path**: leaning-limitless/DuEE-transformers

## Basic Information

- **Project Name**: DuEE-transformers
- **Description**: 基于hugging-face的transformers实现DuEE任务
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 2
- **Forks**: 1
- **Created**: 2020-08-10
- **Last Updated**: 2021-08-18

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README


## 下载预训练模型
下载[chinese-bert_chinese_wwm_pytorch模型](https://github.com/ymcui/Chinese-BERT-wwm)，并将其解压到../../models/

data目录下文件改名（否则transformers无法正确加载模型）：

bert_config.json 改名为 config.json

chinese_wwm_pytorch.bin 改名为 pytorch_model.bin

data目录下文件移动至上一级目录


## 数据预处理

修改配置文件config/duee_role_label_config_bert.json 

config/duee_trigger_label_config_bert.json中的origin_data_dir

执行 duee_process.py 中的 data_preprocess方法 


##  模型训练

### trigger标注模型

python duee.py --config config/duee_trigger_label_config_bert.json --task trigger_label

### role标注模型

python duee.py --config config/duee_role_label_config_bert.json --task role_label

不修改配置的话，模型结果保存在out_put目录下