Hugging Face's logo

aish31
/
Prompt_genre

Text2Text Generation
Transformers
TensorFlow
bart
generated_from_keras_callback
Inference Endpoints
Model card Files Community
aish31 commited on
Commit
63bfda4
•
1 Parent(s): ccba4f0

Upload TFBartForConditionalGeneration

Browse files
Files changed (3) hide show
  1. README.md +13 -13
  2. config.json +1 -1
  3. tf_model.h5 +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: apache-2.0
3
- base_model: aish31/pop_genre5
4
  tags:
5
  - generated_from_keras_callback
6
  model-index:
@@ -13,10 +13,10 @@ probably proofread and complete it, then remove this comment. -->
13
 
14
  # pop_genre5
15
 
16
- This model is a fine-tuned version of [aish31/pop_genre5](https://huggingface.co/aish31/pop_genre5) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.2288
19
- - Validation Loss: 0.5457
20
  - Epoch: 7
21
 
22
  ## Model description
@@ -43,19 +43,19 @@ The following hyperparameters were used during training:
43
 
44
  | Train Loss | Validation Loss | Epoch |
45
  |:----------:|:---------------:|:-----:|
46
- | 0.5345 | 0.4732 | 0 |
47
- | 0.4216 | 0.4548 | 1 |
48
- | 0.3596 | 0.4762 | 2 |
49
- | 0.3226 | 0.4907 | 3 |
50
- | 0.2934 | 0.4899 | 4 |
51
- | 0.2719 | 0.5186 | 5 |
52
- | 0.2487 | 0.5165 | 6 |
53
- | 0.2288 | 0.5457 | 7 |
54
 
55
 
56
  ### Framework versions
57
 
58
  - Transformers 4.33.1
59
- - TensorFlow 2.12.0
60
  - Datasets 2.14.5
61
  - Tokenizers 0.13.3
 
1
  ---
2
  license: apache-2.0
3
+ base_model: facebook/bart-large
4
  tags:
5
  - generated_from_keras_callback
6
  model-index:
 
13
 
14
  # pop_genre5
15
 
16
+ This model is a fine-tuned version of [facebook/bart-large](https://huggingface.co/facebook/bart-large) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.3563
19
+ - Validation Loss: 0.6763
20
  - Epoch: 7
21
 
22
  ## Model description
 
43
 
44
  | Train Loss | Validation Loss | Epoch |
45
  |:----------:|:---------------:|:-----:|
46
+ | 1.3569 | 0.9766 | 0 |
47
+ | 0.9065 | 0.8187 | 1 |
48
+ | 0.7160 | 0.7519 | 2 |
49
+ | 0.5987 | 0.6857 | 3 |
50
+ | 0.5152 | 0.6609 | 4 |
51
+ | 0.4508 | 0.6720 | 5 |
52
+ | 0.4014 | 0.6485 | 6 |
53
+ | 0.3563 | 0.6763 | 7 |
54
 
55
 
56
  ### Framework versions
57
 
58
  - Transformers 4.33.1
59
+ - TensorFlow 2.13.0
60
  - Datasets 2.14.5
61
  - Tokenizers 0.13.3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "aish31/pop_genre5",
3
  "activation_dropout": 0.1,
4
  "activation_function": "gelu",
5
  "add_bias_logits": false,
 
1
  {
2
+ "_name_or_path": "facebook/bart-large",
3
  "activation_dropout": 0.1,
4
  "activation_function": "gelu",
5
  "add_bias_logits": false,
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:99a11e7701e7e305b3d1ae856e7ab7ca16e36cccd3e51b553adacfae0ea26285
3
  size 1625925412
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a160c87bcbc04d4c5c091af1dca81992bc843d08594765677826d6b3e8dca0e
3
  size 1625925412