End of training

README.md

@@ -5,6 +5,9 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
+- precision
+- recall
+- f1
 model-index:
 - name: XLNet-Reddit-Toxic-Comment-Classification
   results: []
@@ -15,11 +18,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 # XLNet-Reddit-Toxic-Comment-Classification
 
-This model is a fine-tuned version of [xlnet-base-cased](https://huggingface.co/xlnet-base-cased) on the Toxic Comment Classification Challenge dataset.
+This model is a fine-tuned version of [xlnet-base-cased](https://huggingface.co/xlnet-base-cased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2615
-- Rmse: 0.2748
-- Accuracy: 0.9245
+- Loss: 0.3448
+- Rmse: 0.2976
+- Accuracy: 0.9115
+- Precision: 0.9153
+- Recall: 0.9261
+- F1: 0.9206
 
 ## Model description
 
@@ -48,13 +54,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Rmse   | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|
-| 0.3672        | 1.0   | 1073 | 0.2615          | 0.2748 | 0.9245   |
-| 0.2441        | 2.0   | 2146 | 0.3464          | 0.2896 | 0.9161   |
-| 0.1726        | 3.0   | 3219 | 0.3880          | 0.2661 | 0.9292   |
-| 0.1146        | 4.0   | 4292 | 0.3287          | 0.2499 | 0.9376   |
-| 0.0831        | 5.0   | 5365 | 0.3672          | 0.2590 | 0.9329   |
+| Training Loss | Epoch | Step | Validation Loss | Rmse   | Accuracy | Precision | Recall | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:---------:|:------:|:------:|
+| 0.7011        | 1.0   | 1073 | 0.6911          | 0.6674 | 0.5545   | 0.5545    | 1.0    | 0.7134 |
+| 0.6829        | 2.0   | 2146 | 0.6936          | 0.7447 | 0.4455   | 0.0       | 0.0    | 0.0    |
+| 0.7004        | 3.0   | 3219 | 0.6876          | 0.6674 | 0.5545   | 0.5545    | 1.0    | 0.7134 |
+| 0.5187        | 4.0   | 4292 | 0.3967          | 0.3202 | 0.8975   | 0.9117    | 0.9025 | 0.9071 |
+| 0.3208        | 5.0   | 5365 | 0.3448          | 0.2976 | 0.9115   | 0.9153    | 0.9261 | 0.9206 |
 
 
 ### Framework versions