search

iryna-kondr / scikit-llm

Seamlessly integrate LLMs into scikit-learn.
https://beastbyte.ai/
MIT License
3.38k stars 275 forks source link

Supporting NER tasks #62

Closed MateusCichelero closed 5 months ago

MateusCichelero commented 1 year ago

Could we add support for named entity recognition tasks to the library? I see that the user interface would not change much from what was applied in the multi-label classification method, with the difference that instead of the entire text input being classified as one or multiple labels, we would have distinct entities from the text being recognized and labeled (the user could also give a semantic list of possible entities to be recognized as input).

OKUA1 commented 1 year ago

Hello @MateusCichelero ,

Thanks for your suggestion. We will evaluate the possibilities of adding NER in the future.

daniau23 commented 1 year ago

Will that of Topic modelling be added too?

OKUA1 commented 1 year ago

@daniau23 maybe, but hard to say at the moment

fvillena commented 6 months ago

You could use https://github.com/plncmm/llmner

iryna-kondr commented 5 months ago

Resolved by #103