FiscalNote/billsum
Viewer • Updated • 23.5k • 8.93k • 55
How to use heskielsvn/test_t5_for_summarization with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("heskielsvn/test_t5_for_summarization")
model = AutoModelForSeq2SeqLM.from_pretrained("heskielsvn/test_t5_for_summarization", device_map="auto")This model is a fine-tuned version of t5-small on the billsum dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen-len |
|---|---|---|---|---|---|---|---|---|
| No log | 1.0 | 62 | 2.8192 | 0.1239 | 0.0326 | 0.1029 | 0.1031 | 19.0 |
| No log | 2.0 | 124 | 2.6080 | 0.1286 | 0.0385 | 0.1065 | 0.1064 | 19.0 |
| No log | 3.0 | 186 | 2.5422 | 0.1302 | 0.0403 | 0.1077 | 0.1077 | 19.0 |
| No log | 4.0 | 248 | 2.5249 | 0.1332 | 0.0426 | 0.1106 | 0.1106 | 19.0 |