Datasetpublic

teo-sanchez/diffusiondb_ner

Description 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.

sourceHugging Facecc-by-3.0updated 4y agoView on Hugging Face

3likes37downloads

Dataset Card

Description

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 following visualization.

  ├── medium/
  │   ├── photography
  │   ├── painting
  │   ├── rendering
  │   └── illustration
  ├── influence/
  │   ├── artist
  │   ├── genre
  │   ├── artwork
  │   └── repository
  ├── light
  ├── color
  ├── composition
  ├── detail
  └── context/
      ├── era
      ├── weather
      └── emotion

Label Scheme

<summary>View label scheme (16 labels for 1 components)</summary>

Component	Labels
`ner`	`color`, `composition`, `context/emotion`, `context/era`, `context/weather`, `detail`, `influence/artist`, `influence/artwork`, `influence/genre`, `influence/repository`, `light`, `medium/illustration`, `medium/painting`, `medium/photography`, `medium/rendering`, `subject`

</details>