search

allenai / allennlp

An open-source NLP research library, built on PyTorch.
http://www.allennlp.org
Apache License 2.0
11.76k stars 2.25k forks source link

Which is the NER model for current master ? #3998

Closed Hacker-VP closed 4 years ago

Hacker-VP commented 4 years ago

I'm using the latest master and for coref I downloaded a 1.3GB bert model that works with outstanding performance.

I'm also in need of a NER model , but all the models I tried have some kind of configuration errors. I think those models are not compatible with the current master. So could you please tell me which NER model works best with the current master ?

dirkgr commented 4 years ago

Like I said in the other issue, the current model URLs are in https://github.com/allenai/allennlp-demo/blob/55b05b36bcd1dd234011760e1597e2410b4a6f09/models.json. The NER model in that list works. I tried it 5 minutes ago. In the next few days I'll make some changes to put these URLs into a more discoverable place.

Also, I'll move and re-train both of the NER models into the allennlp-models repository before we release version 1.0. Sorry for all the churn right now, but we're trying to get it done to have a stable API once 1.0 ships.

dirkgr commented 4 years ago

is there any method for NER using which I can get the named entities as a list or in a simpler way other than the standard predict method ?

I don't know the answer to that. I'm not that familiar with the NER predictors. Is it difficult to go from that output to a list?

Troied commented 4 years ago

@dirkgr Thanks for your help :-)