Hugging Face's logo

prateeksahu147
/
keyword-masked-model

Text2Text Generation
Transformers
TensorFlow
t5
generated_from_keras_callback
Inference Endpoints
Model card Files Community
2

Upload TFT5ForConditionalGeneration

#2
by prateeksahu147 - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+17
-17
Files changed (2) hide show
  1. README.md +16 -16
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -15,12 +15,12 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.9545
19
- - Validation Loss: 0.7767
20
- - Train Rouge1: 76.8067
21
- - Train Rouge2: 64.2138
22
- - Train Rougel: 76.1124
23
- - Train Rougelsum: 76.0843
24
  - Epoch: 9
25
 
26
  ## Model description
@@ -47,16 +47,16 @@ The following hyperparameters were used during training:
47
 
48
  | Train Loss | Validation Loss | Train Rouge1 | Train Rouge2 | Train Rougel | Train Rougelsum | Epoch |
49
  |:----------:|:---------------:|:------------:|:------------:|:------------:|:---------------:|:-----:|
50
- | 2.7609 | 2.0508 | 69.5404 | 56.1190 | 67.9736 | 67.9553 | 0 |
51
- | 2.0227 | 1.5912 | 71.6288 | 58.5009 | 70.6498 | 70.7363 | 1 |
52
- | 1.6780 | 1.3604 | 70.8191 | 58.1082 | 70.0231 | 69.8913 | 2 |
53
- | 1.4974 | 1.2163 | 69.9943 | 57.3099 | 68.7338 | 68.7593 | 3 |
54
- | 1.3525 | 1.1001 | 70.4850 | 57.7869 | 69.2259 | 69.2106 | 4 |
55
- | 1.2480 | 1.0154 | 71.1204 | 58.1709 | 69.9356 | 69.9324 | 5 |
56
- | 1.1556 | 0.9368 | 72.5520 | 59.4597 | 71.3193 | 71.4219 | 6 |
57
- | 1.0820 | 0.8754 | 74.5839 | 61.6499 | 73.6577 | 73.6458 | 7 |
58
- | 1.0117 | 0.8244 | 75.0935 | 62.3254 | 74.3574 | 74.2253 | 8 |
59
- | 0.9545 | 0.7767 | 76.8067 | 64.2138 | 76.1124 | 76.0843 | 9 |
60
 
61
 
62
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.6588
19
+ - Validation Loss: 0.5614
20
+ - Train Rouge1: 81.6702
21
+ - Train Rouge2: 69.0116
22
+ - Train Rougel: 81.6273
23
+ - Train Rougelsum: 81.5364
24
  - Epoch: 9
25
 
26
  ## Model description
 
47
 
48
  | Train Loss | Validation Loss | Train Rouge1 | Train Rouge2 | Train Rougel | Train Rougelsum | Epoch |
49
  |:----------:|:---------------:|:------------:|:------------:|:------------:|:---------------:|:-----:|
50
+ | 0.9026 | 0.7256 | 78.4320 | 65.5502 | 78.2535 | 78.1327 | 0 |
51
+ | 0.8436 | 0.6875 | 79.2603 | 66.4389 | 79.1002 | 79.0620 | 1 |
52
+ | 0.7989 | 0.6597 | 79.8406 | 66.7444 | 79.5641 | 79.5095 | 2 |
53
+ | 0.7739 | 0.6403 | 81.0719 | 68.0576 | 80.8293 | 80.7287 | 3 |
54
+ | 0.7439 | 0.6246 | 81.0565 | 68.0129 | 80.7808 | 80.6909 | 4 |
55
+ | 0.7209 | 0.6135 | 81.1721 | 68.2028 | 80.9586 | 80.8343 | 5 |
56
+ | 0.6962 | 0.5982 | 81.6791 | 68.9723 | 81.5971 | 81.5262 | 6 |
57
+ | 0.6922 | 0.5822 | 81.7266 | 69.0548 | 81.6877 | 81.6085 | 7 |
58
+ | 0.6657 | 0.5696 | 82.0421 | 69.3520 | 81.9003 | 81.8580 | 8 |
59
+ | 0.6588 | 0.5614 | 81.6702 | 69.0116 | 81.6273 | 81.5364 | 9 |
60
 
61
 
62
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1a32409fe8fa8ba11ab3352738c0ddd1027e2fd73f3713d8bd18547dd11a5dc2
3
  size 373902664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a9b1be802a946c2536de3300ae43093ee955cd51da120914567398b9c6c63aa
3
  size 373902664