Downloads · 30 days
35
21% of all-time downloads
derenrich/wiki-kid-friendly-classification-pipeline
wiki-kid-friendly-classification-pipeline is a text classification model from derenrich. Use it when you need a label for a piece of text. It is set up for transformers.
This model takes an English Wikipedia article and tries to decide if the topic is kid-friendly. If it is not it assigns to one of the categories: crime, terrorism, political, mental health, controvserial, disaster, se…
Downloads · 30 days
35
21% of all-time downloads
All-time downloads
165
Public
Parameters
150M
598 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors598 MB · 99%
From the Hugging Face model README
This model takes an English Wikipedia article and tries to decide if the topic is kid-friendly. If it is not it assigns to one of the categories: crime, terrorism, political, mental health, controvserial, disaster, sex, drugs
The model is not vetted for accuracy or bias. Anecdotally it seems to perform well though it may be overly sensistive and it is likely biased. Mainly meant for prototypes where you don't yet know how you need to be filtering before making a more complete model.
Labels for training this model were generated both by human curation and with an LLM.
Use the code below to get started with the model.
>>> from transformers import pipeline
>>> p = pipeline("text-classification", model='derenrich/enwiki-kid-friendly-classifier')
>>> p("Armenian genocide")
[{'label': 'crime', 'score': 0.9999203681945801}]
Daniel Erenrich [email protected]