/OpenNRE

复现 @taorui-plus 的结果

Primary LanguagePythonMIT LicenseMIT

Dataset

The valid dataset is a sub dataset of the trainning dataset, thus the model implemented by @taorui has a 0.97 accuarcy.

Chinese_wwm

Load the pre-train LM

  1. download Chinese_wmm from GitHub
  2. extract the files and paste them to opennre/pretrain/chinese_wwm
  3. rename bert_config.json to config.json

Train the Model

python example/train_supervised_chinese_wwm.py \
    --pretrain_path {os.getcwd()}/OpenNRE/opennre/pretrain/chinese_wwm \
    --dataset people-relation

The ckpt file will save in OpenNRE/ckpt folder.