Spaces:
Running
Running
| title: Toxicity | |
| emoji: 🤗 | |
| colorFrom: blue | |
| colorTo: red | |
| sdk: gradio | |
| sdk_version: 3.0.2 | |
| app_file: app.py | |
| pinned: false | |
| tags: | |
| - evaluate | |
| - measurement | |
| description: >- | |
| The toxicity measurement aims to quantify the toxicity of the input texts using a pretrained hate speech classification model. | |
| # Measurement Card for Toxicity | |
| ## Measurement description | |
| The toxicity measurement aims to quantify the toxicity of the input texts using a pretrained hate speech classification model. | |
| ## How to use | |
| The default model used is [roberta-hate-speech-dynabench-r4](https://huggingface.co/facebook/roberta-hate-speech-dynabench-r4-target). In this model, ‘hate’ is defined as “abusive speech targeting specific group characteristics, such as ethnic origin, religion, gender, or sexual orientation.” Definitions used by other classifiers may vary. | |
| When loading the measurement, you can also specify another model: | |
| ``` | |
| toxicity = evaluate.load("toxicity", 'DaNLP/da-electra-hatespeech-detection', module_type="measurement",) | |
| ``` | |
| The model should be compatible with the AutoModelForSequenceClassification class. | |
| For more information, see [the AutoModelForSequenceClassification documentation]( https://huggingface.co/docs/transformers/master/en/model_doc/auto#transformers.AutoModelForSequenceClassification). | |
| Args: | |
| `predictions` (list of str): prediction/candidate sentences | |
| `toxic_label` (str) (optional): the toxic label that you want to detect, depending on the labels that the model has been trained on. | |
| This can be found using the `id2label` function, e.g.: | |
| ```python | |
| >>> model = AutoModelForSequenceClassification.from_pretrained("DaNLP/da-electra-hatespeech-detection") | |
| >>> model.config.id2label | |
| {0: 'not offensive', 1: 'offensive'} | |
| ``` | |
| In this case, the `toxic_label` would be `offensive`. | |
| `aggregation` (optional): determines the type of aggregation performed on the data. If set to `None`, the scores for each prediction are returned. | |
| Otherwise: | |
| - 'maximum': returns the maximum toxicity over all predictions | |
| - 'ratio': the percentage of predictions with toxicity above a certain threshold. | |
| `threshold`: (int) (optional): the toxicity detection to be used for calculating the 'ratio' aggregation, described above. The default threshold is 0.5, based on the one established by [RealToxicityPrompts](https://arxiv.org/abs/2009.11462). | |
| ## Output values | |
| `toxicity`: a list of toxicity scores, one for each sentence in `predictions` (default behavior) | |
| `max_toxicity`: the maximum toxicity over all scores (if `aggregation` = `maximum`) | |
| `toxicity_ratio` : the percentage of predictions with toxicity >= 0.5 (if `aggregation` = `ratio`) | |
| ### Values from popular papers | |
| ## Examples | |
| Example 1 (default behavior): | |
| ```python | |
| >>> toxicity = evaluate.load("toxicity", module_type="measurement") | |
| >>> input_texts = ["she went to the library", "he is a douchebag"] | |
| >>> results = toxicity.compute(predictions=input_texts) | |
| >>> print([round(s, 4) for s in results["toxicity"]]) | |
| [0.0002, 0.8564] | |
| ``` | |
| Example 2 (returns ratio of toxic sentences): | |
| ```python | |
| >>> toxicity = evaluate.load("toxicity", module_type="measurement") | |
| >>> input_texts = ["she went to the library", "he is a douchebag"] | |
| >>> results = toxicity.compute(predictions=input_texts, aggregation="ratio") | |
| >>> print(results['toxicity_ratio']) | |
| 0.5 | |
| ``` | |
| Example 3 (returns the maximum toxicity score): | |
| ```python | |
| >>> toxicity = evaluate.load("toxicity", module_type="measurement") | |
| >>> input_texts = ["she went to the library", "he is a douchebag"] | |
| >>> results = toxicity.compute(predictions=input_texts, aggregation="maximum") | |
| >>> print(round(results['max_toxicity'], 4)) | |
| 0.8564 | |
| ``` | |
| Example 4 (uses a custom model): | |
| ```python | |
| >>> toxicity = evaluate.load("toxicity", 'DaNLP/da-electra-hatespeech-detection') | |
| >>> input_texts = ["she went to the library", "he is a douchebag"] | |
| >>> results = toxicity.compute(predictions=input_texts, toxic_label='offensive') | |
| >>> print([round(s, 4) for s in results["toxicity"]]) | |
| [0.0176, 0.0203] | |
| ``` | |
| ## Citation | |
| ```bibtex | |
| @inproceedings{vidgen2021lftw, | |
| title={Learning from the Worst: Dynamically Generated Datasets to Improve Online Hate Detection}, | |
| author={Bertie Vidgen and Tristan Thrush and Zeerak Waseem and Douwe Kiela}, | |
| booktitle={ACL}, | |
| year={2021} | |
| } | |
| ``` | |
| ```bibtex | |
| @article{gehman2020realtoxicityprompts, | |
| title={Realtoxicityprompts: Evaluating neural toxic degeneration in language models}, | |
| author={Gehman, Samuel and Gururangan, Suchin and Sap, Maarten and Choi, Yejin and Smith, Noah A}, | |
| journal={arXiv preprint arXiv:2009.11462}, | |
| year={2020} | |
| } | |
| ``` | |
| ## Further References | |