代码拉取完成,页面将自动刷新
下载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方法
python duee.py --config config/duee_trigger_label_config_bert.json --task trigger_label
python duee.py --config config/duee_role_label_config_bert.json --task role_label
不修改配置的话,模型结果保存在out_put目录下
你可以在登录后,发表评论
仓库评论 ( 0 )