devhem
/

LLMGUARD-roberta

Text Classification

Generated from Trainer

Model card Files Files and versions Community

devhem commited on Jan 9

Commit

368fd13

·

verified ·

1 Parent(s): f3f886c

Model save

Files changed (1) hide show

README.md +6 -6

README.md CHANGED Viewed

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6789
-- Accuracy: 0.7720
 ## Model description
@@ -52,10 +52,10 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.1665        | 1.0   | 1332 | 0.7611          | 0.7564   |
-| 0.6955        | 2.0   | 2664 | 0.7009          | 0.7681   |
-| 0.6476        | 3.0   | 3996 | 0.6831          | 0.7743   |
-| 0.6155        | 4.0   | 5328 | 0.6789          | 0.7720   |
 ### Framework versions

 This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.6480
+- Accuracy: 0.7775
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 1.1863        | 1.0   | 1332 | 0.7541          | 0.7598   |
+| 0.7092        | 2.0   | 2664 | 0.6746          | 0.7690   |
+| 0.6626        | 3.0   | 3996 | 0.6527          | 0.7760   |
+| 0.6266        | 4.0   | 5328 | 0.6480          | 0.7775   |
 ### Framework versions