Hugging Face's logo

minh21
/
XLNet-Reddit-Toxic-Comment-Classification

Text Classification
Transformers
PyTorch
xlnet
Generated from Trainer
Inference Endpoints
Model card Files Community
minh21 commited on
Commit
bf47c7f
•
1 Parent(s): 429b0ab

End of training

Browse files
Files changed (2) hide show
  1. README.md +11 -11
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -20,12 +20,12 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [xlnet-base-cased](https://huggingface.co/xlnet-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.2248
24
- - Rmse: 0.2928
25
- - Accuracy: 0.9143
26
- - Precision: 0.9299
27
- - Recall: 0.9143
28
- - F1: 0.9220
29
 
30
  ## Model description
31
 
@@ -56,11 +56,11 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Rmse | Accuracy | Precision | Recall | F1 |
58
  |:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:---------:|:------:|:------:|
59
- | 0.3656 | 1.0 | 1073 | 0.2248 | 0.2928 | 0.9143 | 0.9299 | 0.9143 | 0.9220 |
60
- | 0.2432 | 2.0 | 2146 | 0.3105 | 0.2912 | 0.9152 | 0.9158 | 0.9328 | 0.9242 |
61
- | 0.1649 | 3.0 | 3219 | 0.3818 | 0.2696 | 0.9273 | 0.9176 | 0.9546 | 0.9357 |
62
- | 0.1075 | 4.0 | 4292 | 0.4398 | 0.2798 | 0.9217 | 0.9049 | 0.9597 | 0.9315 |
63
- | 0.0788 | 5.0 | 5365 | 0.4655 | 0.2847 | 0.9189 | 0.9110 | 0.9462 | 0.9283 |
64
 
65
 
66
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [xlnet-base-cased](https://huggingface.co/xlnet-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.2964
24
+ - Rmse: 0.2828
25
+ - Accuracy: 0.92
26
+ - Precision: 0.9236
27
+ - Recall: 0.9329
28
+ - F1: 0.9282
29
 
30
  ## Model description
31
 
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Rmse | Accuracy | Precision | Recall | F1 |
58
  |:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:---------:|:------:|:------:|
59
+ | 0.3798 | 1.0 | 1075 | 0.2964 | 0.2828 | 0.92 | 0.9236 | 0.9329 | 0.9282 |
60
+ | 0.2507 | 2.0 | 2150 | 0.3791 | 0.2973 | 0.9116 | 0.8824 | 0.9698 | 0.9241 |
61
+ | 0.1734 | 3.0 | 3225 | 0.3779 | 0.3080 | 0.9051 | 0.8847 | 0.9530 | 0.9176 |
62
+ | 0.1157 | 4.0 | 4300 | 0.4796 | 0.2861 | 0.9181 | 0.9456 | 0.9044 | 0.9245 |
63
+ | 0.0762 | 5.0 | 5375 | 0.4729 | 0.2762 | 0.9237 | 0.9341 | 0.9279 | 0.9310 |
64
 
65
 
66
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:db0b55d5e1e0984278c691b6eaeb08810e4cb4ba4691bbab9c4477690d46a518
3
  size 469311765
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f7386724b14f7c8ffd0f8be432d1d5ed4b7465a40414774d1f735df4f1f872d
3
  size 469311765