Magicearth
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -18,6 +18,7 @@ Catégories: toxic, severe_toxic, obscene, threat, insult, identity_hate
|
|
18 |
|
19 |
Le finetuning a été fait pour 4 époques. La dataset utilisé est celui de Google appelé jigsaw_toxicity_pred.
|
20 |
|
|
|
21 |
training_args = TrainingArguments(
|
22 |
output_dir="./results",
|
23 |
evaluation_strategy="epoch",
|
@@ -33,7 +34,7 @@ training_args = TrainingArguments(
|
|
33 |
load_best_model_at_end=True,
|
34 |
)
|
35 |
|
36 |
-
Erreur moyenne absolue par
|
37 |
|
38 |
toxic: 0.0271
|
39 |
severe_toxic: 0.0128
|
|
|
18 |
|
19 |
Le finetuning a été fait pour 4 époques. La dataset utilisé est celui de Google appelé jigsaw_toxicity_pred.
|
20 |
|
21 |
+
# Paramètres d'entraînement
|
22 |
training_args = TrainingArguments(
|
23 |
output_dir="./results",
|
24 |
evaluation_strategy="epoch",
|
|
|
34 |
load_best_model_at_end=True,
|
35 |
)
|
36 |
|
37 |
+
# Erreur moyenne absolue par catégorie sur le dataset d'entraînement:
|
38 |
|
39 |
toxic: 0.0271
|
40 |
severe_toxic: 0.0128
|