dmis-lab / bern

A neural named entity recognition and multi-type normalization tool for biomedical text mining
https://bern.korea.ac.kr
BSD 2-Clause "Simplified" License
173 stars 44 forks source link

Switching off GNormPlus and tmVar; customizing BERN #11

Open dimitrijenyc opened 4 years ago

dimitrijenyc commented 4 years ago

Is there an easy way to turn off or not install at all some of the modules used in NER in BERN? For example, I'm only interested in drug/chemical discovery and I want to skip using GNormPlus and tmVar. Thanks!

donghyeonk commented 4 years ago

First of all, leave only 'drug' in the list on the following line.

https://github.com/dmis-lab/bern/blob/633125980ceeed111942b8c247cf72aca533b6be/biobert_ner/run_ner.py#L402