Extended dataset infered by the name entity recognition model en_ner_prompting. This model has been trained on hand-annotated prompts from poloclub/diffusiondb.
This dataset is hence infered by this model and can comprise mistakes, especially on certain categories (cf. model card).
The entities comprise 7 main categories and 11 subcategories for a total of 16 categories, extracted from a topic analysis made with BERTopic.
The topic analysis can be explored the… See the full description on the dataset page:
https://huggingface.co/datasets/teo-sanchez/diffusiondb_ner.
