update model card README.md
Browse files
README.md
ADDED
@@ -0,0 +1,95 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
tags:
|
4 |
+
- generated_from_trainer
|
5 |
+
metrics:
|
6 |
+
- rouge
|
7 |
+
model-index:
|
8 |
+
- name: BART_pretrained_on_billsum_finetuned_on_small_SCOTUS_extracted_dataset_3
|
9 |
+
results: []
|
10 |
+
---
|
11 |
+
|
12 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
13 |
+
should probably proofread and complete it, then remove this comment. -->
|
14 |
+
|
15 |
+
# BART_pretrained_on_billsum_finetuned_on_small_SCOTUS_extracted_dataset_3
|
16 |
+
|
17 |
+
This model is a fine-tuned version of [bheshaj/bart-large-billsum-epochs20](https://huggingface.co/bheshaj/bart-large-billsum-epochs20) on an unknown dataset.
|
18 |
+
It achieves the following results on the evaluation set:
|
19 |
+
- Loss: 3.5827
|
20 |
+
- Rouge1: 0.1251
|
21 |
+
- Rouge2: 0.037
|
22 |
+
- Rougel: 0.1002
|
23 |
+
- Rougelsum: 0.1003
|
24 |
+
- Gen Len: 19.4207
|
25 |
+
|
26 |
+
## Model description
|
27 |
+
|
28 |
+
More information needed
|
29 |
+
|
30 |
+
## Intended uses & limitations
|
31 |
+
|
32 |
+
More information needed
|
33 |
+
|
34 |
+
## Training and evaluation data
|
35 |
+
|
36 |
+
More information needed
|
37 |
+
|
38 |
+
## Training procedure
|
39 |
+
|
40 |
+
### Training hyperparameters
|
41 |
+
|
42 |
+
The following hyperparameters were used during training:
|
43 |
+
- learning_rate: 2e-05
|
44 |
+
- train_batch_size: 8
|
45 |
+
- eval_batch_size: 8
|
46 |
+
- seed: 42
|
47 |
+
- gradient_accumulation_steps: 8
|
48 |
+
- total_train_batch_size: 64
|
49 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
50 |
+
- lr_scheduler_type: linear
|
51 |
+
- num_epochs: 30
|
52 |
+
- mixed_precision_training: Native AMP
|
53 |
+
|
54 |
+
### Training results
|
55 |
+
|
56 |
+
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
|
57 |
+
|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
|
58 |
+
| 6.7036 | 0.98 | 10 | 4.9345 | 0.1041 | 0.0255 | 0.0865 | 0.0864 | 20.0 |
|
59 |
+
| 4.9048 | 1.98 | 20 | 4.4400 | 0.1201 | 0.0283 | 0.0973 | 0.0974 | 19.9573 |
|
60 |
+
| 4.544 | 2.98 | 30 | 4.2661 | 0.1272 | 0.0298 | 0.1013 | 0.1007 | 19.9695 |
|
61 |
+
| 4.3585 | 3.98 | 40 | 4.1790 | 0.1257 | 0.0289 | 0.1007 | 0.1006 | 19.8293 |
|
62 |
+
| 4.226 | 4.98 | 50 | 4.1156 | 0.1229 | 0.0302 | 0.0994 | 0.0994 | 19.6098 |
|
63 |
+
| 4.1417 | 5.98 | 60 | 4.0503 | 0.1225 | 0.0293 | 0.0985 | 0.0987 | 19.6037 |
|
64 |
+
| 4.0406 | 6.98 | 70 | 3.9856 | 0.1208 | 0.0309 | 0.0975 | 0.0976 | 19.6159 |
|
65 |
+
| 3.9409 | 7.98 | 80 | 3.9266 | 0.1231 | 0.0299 | 0.0982 | 0.0984 | 19.5976 |
|
66 |
+
| 3.8492 | 8.98 | 90 | 3.8767 | 0.1212 | 0.0299 | 0.0953 | 0.0954 | 19.6037 |
|
67 |
+
| 3.7571 | 9.98 | 100 | 3.8241 | 0.1196 | 0.0313 | 0.097 | 0.0972 | 19.811 |
|
68 |
+
| 3.6975 | 10.98 | 110 | 3.7959 | 0.121 | 0.0303 | 0.0963 | 0.0963 | 19.6768 |
|
69 |
+
| 3.5923 | 11.98 | 120 | 3.7628 | 0.115 | 0.0315 | 0.0959 | 0.0962 | 19.7012 |
|
70 |
+
| 3.5505 | 12.98 | 130 | 3.7352 | 0.1166 | 0.034 | 0.0952 | 0.0957 | 19.6829 |
|
71 |
+
| 3.5027 | 13.98 | 140 | 3.7157 | 0.1222 | 0.0347 | 0.1004 | 0.1005 | 19.6341 |
|
72 |
+
| 3.456 | 14.98 | 150 | 3.6983 | 0.1198 | 0.032 | 0.0968 | 0.097 | 19.6524 |
|
73 |
+
| 3.4088 | 15.98 | 160 | 3.6644 | 0.1204 | 0.0321 | 0.0969 | 0.0969 | 19.4695 |
|
74 |
+
| 3.3511 | 16.98 | 170 | 3.6545 | 0.1224 | 0.035 | 0.1001 | 0.1004 | 19.8171 |
|
75 |
+
| 3.3167 | 17.98 | 180 | 3.6415 | 0.1223 | 0.0363 | 0.1006 | 0.1007 | 19.7683 |
|
76 |
+
| 3.2786 | 18.98 | 190 | 3.6286 | 0.1234 | 0.0345 | 0.1004 | 0.1005 | 19.9756 |
|
77 |
+
| 3.2437 | 19.98 | 200 | 3.6239 | 0.124 | 0.0368 | 0.1006 | 0.1009 | 19.6829 |
|
78 |
+
| 3.2114 | 20.98 | 210 | 3.6138 | 0.1256 | 0.0393 | 0.1035 | 0.104 | 19.7256 |
|
79 |
+
| 3.1935 | 21.98 | 220 | 3.6025 | 0.1241 | 0.0359 | 0.1016 | 0.1016 | 19.5122 |
|
80 |
+
| 3.175 | 22.98 | 230 | 3.5939 | 0.1213 | 0.0356 | 0.1008 | 0.1011 | 19.4024 |
|
81 |
+
| 3.1572 | 23.98 | 240 | 3.5979 | 0.124 | 0.0355 | 0.1008 | 0.1007 | 19.7256 |
|
82 |
+
| 3.1346 | 24.98 | 250 | 3.5909 | 0.1247 | 0.0356 | 0.1011 | 0.1014 | 19.6037 |
|
83 |
+
| 3.1202 | 25.98 | 260 | 3.5877 | 0.1299 | 0.0372 | 0.1042 | 0.1045 | 19.7866 |
|
84 |
+
| 3.1095 | 26.98 | 270 | 3.5876 | 0.13 | 0.0381 | 0.1056 | 0.1059 | 19.7866 |
|
85 |
+
| 3.0919 | 27.98 | 280 | 3.5851 | 0.1286 | 0.04 | 0.1037 | 0.1037 | 19.6951 |
|
86 |
+
| 3.1089 | 28.98 | 290 | 3.5830 | 0.1253 | 0.0376 | 0.1001 | 0.1002 | 19.4207 |
|
87 |
+
| 3.0915 | 29.98 | 300 | 3.5827 | 0.1251 | 0.037 | 0.1002 | 0.1003 | 19.4207 |
|
88 |
+
|
89 |
+
|
90 |
+
### Framework versions
|
91 |
+
|
92 |
+
- Transformers 4.26.1
|
93 |
+
- Pytorch 1.13.1+cu116
|
94 |
+
- Datasets 2.10.1
|
95 |
+
- Tokenizers 0.13.2
|