1 Star 2 Fork 1

learning-limitless / DuEE-transformers

Create your Gitee Account
Explore and code with more than 8 million developers,Free private repositories !:)
Sign up
This repository doesn't specify license. Please pay attention to the specific project description and its upstream code dependency when using it.
Clone or Download
Cancel
Notice: Creating folder will generate an empty file .keep, because not support in Git
Loading...
readme.md

下载预训练模型

下载chinese-bert_chinese_wwm_pytorch模型,并将其解压到../../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目录下

Repository Comments ( 0 )

Sign in to post a comment

About

基于hugging-face的transformers实现DuEE任务 expand collapse
Cancel

Releases

No release

Contributors

All

Activities

Load More
can not load any more
Python
1
https://gitee.com/leaning-limitless/DuEE-transformers.git
git@gitee.com:leaning-limitless/DuEE-transformers.git
leaning-limitless
DuEE-transformers
DuEE-transformers
master

Search