search

allenai / scibert

A BERT model for scientific text.
https://arxiv.org/abs/1903.10676
Apache License 2.0
1.47k stars 214 forks source link

No default predictor for model type bert_crf_tagger. #102

Open dpappas opened 3 years ago

dpappas commented 3 years ago

Hi everyone!

I tried the following command to use my fine-tuned model for NER.

python -m allennlp.run predict --output-file=out.txt --include-package=scibert ./ner_model_bc5cdr/model.tar.gz ./data/ner/bc5cdr/test.txt

I have used the same commands you mention in your readme file and a virtual environment so the versions I use are exactly the same as yours.

I get the following error: 'No default predictor for model type bert_crf_tagger.\nPlease specify a predictor explicitly.'

How can I overcome this problem? Also, how do I specify the predictor? I tried --predictor=ner_model_bc5cdr --predictor=./scibert/models/bert_crf_tagger --predictor=./scibert/models/bert_crf_tagger.py but nothing worked.

Can you help me?

aadarshsingh191198 commented 3 years ago

--predictor sentence-tagger is probably what you are looking for