Update README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,7 @@ language:
|
|
6 |
# Text Classification Toxicity
|
7 |
|
8 |
This model is a fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large) on the on the [Jigsaw 1st Kaggle competition](https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge) dataset using [unitary/toxic-bert](https://huggingface.co/unitary/toxic-bert) as teacher model.
|
9 |
-
The
|
10 |
|
11 |
The model contains two labels only (toxicity and severe toxicity). For the model with all labels refer to this [page](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw)
|
12 |
|
@@ -93,7 +93,7 @@ for result in results:
|
|
93 |
res.append(max_score)
|
94 |
|
95 |
res
|
96 |
-
# [('toxic', 0.
|
97 |
```
|
98 |
|
99 |
# Training hyperparameters
|
|
|
6 |
# Text Classification Toxicity
|
7 |
|
8 |
This model is a fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large) on the on the [Jigsaw 1st Kaggle competition](https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge) dataset using [unitary/toxic-bert](https://huggingface.co/unitary/toxic-bert) as teacher model.
|
9 |
+
The original unquantized model can be found here [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw-lite).
|
10 |
|
11 |
The model contains two labels only (toxicity and severe toxicity). For the model with all labels refer to this [page](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw)
|
12 |
|
|
|
93 |
res.append(max_score)
|
94 |
|
95 |
res
|
96 |
+
# [('toxic', 0.6553249955177307)]
|
97 |
```
|
98 |
|
99 |
# Training hyperparameters
|