请问可以应用在中文的语料库吗 about kg-bert HOT 2 CLOSED

@WenTingTseng

你好，

您在调用代码时将预训练BERT模型从英文的bert-base-uncased和bert-base-cased换成中文的'bert-base-chinese即可, 即在命令

python3 run_bert_relation_prediction.py
--task_name kg
--do_train
--do_eval
--do_predict
--data_dir ./data/XXX
--bert_model bert-base-cased
--max_seq_length 25
--train_batch_size 32
--learning_rate 5e-5
--num_train_epochs 20.0
--output_dir ./output_FB15K/
--gradient_accumulation_steps 1
--eval_batch_size 512

中使用：

--bert_model bert-base-chinese，而不是 --bert_model bert-base-cased

中文知识图谱可参考：https://github.com/ownthink/KnowledgeGraphData

这篇论文之前投的会议，没中，现在准备转投IEEE Transactions期刊，可能PAMI，或TKDE。

from kg-bert.

好的,了解

谢谢你

from kg-bert.