laureanadcastro
/

mbart-neutralization

Text2Text Generation

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

laureanadcastro commited on Feb 16

Commit

a142269

•

1 Parent(s): a8c58d4

Training complete

Files changed (2) hide show

README.md +5 -5
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 5.2526
-- Bleu: 20.3083
-- Gen Len: 12.7561
 ## Model description
@@ -51,8 +51,8 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Bleu    | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
-| No log        | 1.0   | 16   | 5.6569          | 18.8367 | 12.4878 |
-| No log        | 2.0   | 32   | 5.2526          | 20.3083 | 12.7561 |
 ### Framework versions

 This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.0181
+- Bleu: 98.7341
+- Gen Len: 18.4896
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Bleu    | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
+| No log        | 1.0   | 440  | 0.0307          | 91.2911 | 18.25   |
+| 0.2343        | 2.0   | 880  | 0.0181          | 98.7341 | 18.4896 |
 ### Framework versions

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:82b99059e8a57dff52042a00e25d4bad26516a7860a6c6a7eec46d08d1ec4894
 size 2444578688

 version https://git-lfs.github.com/spec/v1
+oid sha256:be81fdab8cada8f8cdd8d57dfd898b9eeb36743291ee9865ff988dadb47b2efc
 size 2444578688