autoliuweijie / K-BERT

Source code of K-BERT (AAAI2020)
https://ojs.aaai.org//index.php/AAAI/article/view/5681
949 stars 212 forks source link

Prediction #48

Open Kiaramarnitt27 opened 3 years ago

Kiaramarnitt27 commented 3 years ago

I have run the run_kbert_ner.py script with the Medicine_NER data. I have obtained the fine-tuned model. However, I was wondering to do the inference to obtain the prediction. I have seen in the UER repository, that there is an additional folder with the inference scripts "inference/run_ner_infer.py" How does this work for your code?