Hugging Face's logo

aiguy68
/
Super_legal_text_summarizer

Text2Text Generation
Transformers
TensorBoard
Safetensors
bart
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
aiguy68 commited on
Commit
43b7224
•
1 Parent(s): fbc2dc9

End of training

Browse files
Files changed (3) hide show
  1. README.md +15 -15
  2. generation_config.json +12 -3
  3. model.safetensors +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
- license: apache-2.0
3
- base_model: google-t5/t5-small
4
  tags:
5
  - generated_from_trainer
6
  metrics:
@@ -15,14 +15,14 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # Super_legal_text_summarizer
17
 
18
- This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 3.0471
21
- - Rouge1: 0.1436
22
- - Rouge2: 0.0449
23
- - Rougel: 0.115
24
- - Rougelsum: 0.115
25
- - Gen Len: 19.0
26
 
27
  ## Model description
28
 
@@ -52,12 +52,12 @@ The following hyperparameters were used during training:
52
 
53
  ### Training results
54
 
55
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
56
- |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
57
- | No log | 1.0 | 203 | 3.2636 | 0.1254 | 0.04 | 0.1009 | 0.1008 | 19.0 |
58
- | No log | 2.0 | 406 | 3.1259 | 0.1392 | 0.045 | 0.112 | 0.1116 | 19.0 |
59
- | 3.4985 | 3.0 | 609 | 3.0658 | 0.1433 | 0.0446 | 0.1145 | 0.1146 | 19.0 |
60
- | 3.4985 | 4.0 | 812 | 3.0471 | 0.1436 | 0.0449 | 0.115 | 0.115 | 19.0 |
61
 
62
 
63
  ### Framework versions
 
1
  ---
2
+ license: mit
3
+ base_model: facebook/bart-large-cnn
4
  tags:
5
  - generated_from_trainer
6
  metrics:
 
15
 
16
  # Super_legal_text_summarizer
17
 
18
+ This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 1.8504
21
+ - Rouge1: 0.4106
22
+ - Rouge2: 0.1827
23
+ - Rougel: 0.2604
24
+ - Rougelsum: 0.2624
25
+ - Gen Len: 130.9261
26
 
27
  ## Model description
28
 
 
52
 
53
  ### Training results
54
 
55
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
56
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:--------:|
57
+ | No log | 1.0 | 203 | 1.9295 | 0.4004 | 0.1792 | 0.2546 | 0.2562 | 123.1576 |
58
+ | No log | 2.0 | 406 | 1.8438 | 0.4163 | 0.1886 | 0.2607 | 0.2625 | 125.9655 |
59
+ | 1.8737 | 3.0 | 609 | 1.8503 | 0.4044 | 0.1721 | 0.2498 | 0.2512 | 132.9951 |
60
+ | 1.8737 | 4.0 | 812 | 1.8504 | 0.4106 | 0.1827 | 0.2604 | 0.2624 | 130.9261 |
61
 
62
 
63
  ### Framework versions
generation_config.json CHANGED
@@ -1,6 +1,15 @@
1
  {
2
- "decoder_start_token_id": 0,
3
- "eos_token_id": 1,
4
- "pad_token_id": 0,
 
 
 
 
 
 
 
 
 
5
  "transformers_version": "4.39.3"
6
  }
 
1
  {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 2,
4
+ "early_stopping": true,
5
+ "eos_token_id": 2,
6
+ "forced_bos_token_id": 0,
7
+ "forced_eos_token_id": 2,
8
+ "length_penalty": 2.0,
9
+ "max_length": 142,
10
+ "min_length": 56,
11
+ "no_repeat_ngram_size": 3,
12
+ "num_beams": 4,
13
+ "pad_token_id": 1,
14
  "transformers_version": "4.39.3"
15
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7d93f45f1018f420398d03eaeb28e98447c4b51269e7957600a368a9d1a029d9
3
  size 1625422896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0228c0ad029ab668cdd9755b2ca220b3c776c6045de6a44172fa3f48cd18fc15
3
  size 1625422896