annahaz
/

xlm-roberta-base-misogyny-sexism-indomain-mix-bal

Text Classification

Inference Endpoints

Model card Files Files and versions Community

annahaz commited on Aug 17, 2022

Commit

98f6a8c

•

1 Parent(s): 8d00ce1

update model card README.md

Files changed (1) hide show

README.md +15 -17

README.md CHANGED Viewed

@@ -19,16 +19,16 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6715
-- Accuracy: 0.802
-- F1: 0.7735
-- Precision: 0.9037
-- Recall: 0.676
-- Mae: 0.198
-- Tn: 464
-- Fp: 36
-- Fn: 162
-- Tp: 338
 ## Model description
@@ -53,16 +53,14 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
 ### Training results
-| Training Loss | Epoch | Step  | Validation Loss | Accuracy | F1     | Precision | Recall | Mae   | Tn  | Fp | Fn  | Tp  |
-|:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|:-----:|:---:|:--:|:---:|:---:|
-| 0.3727        | 1.0   | 2714  | 0.6816          | 0.735    | 0.6683 | 0.8930    | 0.534  | 0.265 | 468 | 32 | 233 | 267 |
-| 0.3257        | 2.0   | 5428  | 0.6787          | 0.753    | 0.6893 | 0.9288    | 0.548  | 0.247 | 479 | 21 | 226 | 274 |
-| 0.2785        | 3.0   | 8142  | 0.5640          | 0.779    | 0.7397 | 0.8997    | 0.628  | 0.221 | 465 | 35 | 186 | 314 |
-| 0.25          | 4.0   | 10856 | 0.6715          | 0.802    | 0.7735 | 0.9037    | 0.676  | 0.198 | 464 | 36 | 162 | 338 |
 ### Framework versions

 This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.8259
+- Accuracy: 0.826
+- F1: 0.8333
+- Precision: 0.7996
+- Recall: 0.87
+- Mae: 0.174
+- Tn: 391
+- Fp: 109
+- Fn: 65
+- Tp: 435
 ## Model description
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
+- num_epochs: 2
 ### Training results
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall | Mae   | Tn  | Fp  | Fn | Tp  |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|:-----:|:---:|:---:|:--:|:---:|
+| 0.2643        | 1.0   | 1603 | 0.6511          | 0.82     | 0.8269 | 0.7963    | 0.86   | 0.18  | 390 | 110 | 70 | 430 |
+| 0.2004        | 2.0   | 3206 | 0.8259          | 0.826    | 0.8333 | 0.7996    | 0.87   | 0.174 | 391 | 109 | 65 | 435 |
 ### Framework versions