Hugging Face's logo

kaifanli
/
bart-base-japanese-tobyoki-pairwise

Text2Text Generation
Transformers
PyTorch
TensorBoard
Safetensors
mbart
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
kaifanli commited on
Commit
98f4bdc
•
1 Parent(s): eefb311

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -13
README.md CHANGED
@@ -16,12 +16,12 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [ku-nlp/bart-base-japanese](https://huggingface.co/ku-nlp/bart-base-japanese) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 3.3866
20
- - Rouge1: 11.5579
21
- - Rouge2: 2.3617
22
- - Rougel: 8.299
23
- - Rougelsum: 9.8507
24
- - Gen Len: 43.3687
25
 
26
  ## Model description
27
 
@@ -40,7 +40,7 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - learning_rate: 5e-05
44
  - train_batch_size: 1
45
  - eval_batch_size: 1
46
  - seed: 42
@@ -50,12 +50,13 @@ The following hyperparameters were used during training:
50
 
51
  ### Training results
52
 
53
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
54
- |:-------------:|:-----:|:-----:|:---------------:|:-------:|:------:|:------:|:---------:|:-------:|
55
- | 0.298 | 1.0 | 4332 | 2.7362 | 11.716 | 1.7738 | 7.7212 | 10.0982 | 52.1582 |
56
- | 0.1596 | 2.0 | 8664 | 3.1080 | 11.6339 | 1.9589 | 8.2226 | 9.5881 | 42.7579 |
57
- | 0.0805 | 3.0 | 12996 | 3.2878 | 11.3717 | 2.0083 | 7.8703 | 9.6383 | 43.2468 |
58
- | 0.0576 | 4.0 | 17328 | 3.3866 | 11.5579 | 2.3617 | 8.299 | 9.8507 | 43.3687 |
 
59
 
60
 
61
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [ku-nlp/bart-base-japanese](https://huggingface.co/ku-nlp/bart-base-japanese) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 2.3352
20
+ - Rouge1: 2.9535
21
+ - Rouge2: 0.589
22
+ - Rougel: 2.0958
23
+ - Rougelsum: 2.7622
24
+ - Gen Len: 15.7896
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 3e-06
44
  - train_batch_size: 1
45
  - eval_batch_size: 1
46
  - seed: 42
 
50
 
51
  ### Training results
52
 
53
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
54
+ |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
55
+ | 1.9659 | 1.0 | 4332 | 2.3037 | 1.596 | 0.3213 | 1.1975 | 1.4562 | 8.4066 |
56
+ | 1.555 | 2.0 | 8664 | 2.2936 | 3.1408 | 0.6522 | 2.3125 | 2.982 | 12.3861 |
57
+ | 1.3883 | 3.0 | 12996 | 2.3163 | 2.6939 | 0.5669 | 1.9944 | 2.6112 | 11.1440 |
58
+ | 1.2592 | 4.0 | 17328 | 2.3362 | 2.3473 | 0.4205 | 1.6502 | 2.2055 | 12.8228 |
59
+ | 1.2024 | 5.0 | 21660 | 2.3352 | 2.9535 | 0.589 | 2.0958 | 2.7622 | 15.7896 |
60
 
61
 
62
  ### Framework versions