1 Star 2 Fork 1

learning-limitless / DuEE-transformers

加入 Gitee
与超过 800 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
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目录下

仓库评论 ( 0 )

你可以在登录后,发表评论

简介

基于hugging-face的transformers实现DuEE任务 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
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

搜索帮助