Commit
•
9bb6a4a
1
Parent(s):
6e97d50
Update README.md
Browse files
README.md
CHANGED
|
@@ -13,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 13 |
|
| 14 |
# toxicity-classification-multi-class
|
| 15 |
|
| 16 |
-
This model is a fine-tuned version of [line-corporation/line-distilbert-base-japanese](https://huggingface.co/line-corporation/line-distilbert-base-japanese) on the
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
- Loss: 0.2642
|
| 19 |
- Roc Auc: 0.8447
|
|
@@ -43,32 +43,6 @@ The following hyperparameters were used during training:
|
|
| 43 |
- lr_scheduler_type: linear
|
| 44 |
- num_epochs: 20
|
| 45 |
|
| 46 |
-
### Training results
|
| 47 |
-
|
| 48 |
-
| Training Loss | Epoch | Step | Validation Loss | Roc Auc |
|
| 49 |
-
|:-------------:|:-----:|:----:|:---------------:|:-------:|
|
| 50 |
-
| No log | 1.0 | 20 | 0.5432 | 0.4938 |
|
| 51 |
-
| No log | 2.0 | 40 | 0.4458 | 0.5062 |
|
| 52 |
-
| No log | 3.0 | 60 | 0.3768 | 0.5753 |
|
| 53 |
-
| No log | 4.0 | 80 | 0.3401 | 0.6401 |
|
| 54 |
-
| No log | 5.0 | 100 | 0.3228 | 0.7297 |
|
| 55 |
-
| No log | 6.0 | 120 | 0.3134 | 0.7894 |
|
| 56 |
-
| No log | 7.0 | 140 | 0.3061 | 0.8185 |
|
| 57 |
-
| No log | 8.0 | 160 | 0.3000 | 0.8331 |
|
| 58 |
-
| No log | 9.0 | 180 | 0.2949 | 0.8451 |
|
| 59 |
-
| No log | 10.0 | 200 | 0.2864 | 0.8434 |
|
| 60 |
-
| No log | 11.0 | 220 | 0.2794 | 0.8473 |
|
| 61 |
-
| No log | 12.0 | 240 | 0.2748 | 0.8481 |
|
| 62 |
-
| No log | 13.0 | 260 | 0.2724 | 0.8404 |
|
| 63 |
-
| No log | 14.0 | 280 | 0.2685 | 0.8400 |
|
| 64 |
-
| No log | 15.0 | 300 | 0.2668 | 0.8451 |
|
| 65 |
-
| No log | 16.0 | 320 | 0.2670 | 0.8481 |
|
| 66 |
-
| No log | 17.0 | 340 | 0.2651 | 0.8430 |
|
| 67 |
-
| No log | 18.0 | 360 | 0.2651 | 0.8434 |
|
| 68 |
-
| No log | 19.0 | 380 | 0.2642 | 0.8434 |
|
| 69 |
-
| No log | 20.0 | 400 | 0.2642 | 0.8447 |
|
| 70 |
-
|
| 71 |
-
|
| 72 |
### Framework versions
|
| 73 |
|
| 74 |
- Transformers 4.34.0
|
|
|
|
| 13 |
|
| 14 |
# toxicity-classification-multi-class
|
| 15 |
|
| 16 |
+
This model is a fine-tuned version of [line-corporation/line-distilbert-base-japanese](https://huggingface.co/line-corporation/line-distilbert-base-japanese) on the [Japanese toxicity dataset](https://github.com/inspection-ai/japanese-toxic-dataset/tree/main).
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
- Loss: 0.2642
|
| 19 |
- Roc Auc: 0.8447
|
|
|
|
| 43 |
- lr_scheduler_type: linear
|
| 44 |
- num_epochs: 20
|
| 45 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
### Framework versions
|
| 47 |
|
| 48 |
- Transformers 4.34.0
|