defunct-datasets/amazon_reviews_multi
Updated • 4.53k • 102
How to use Hoax0930/summary_tutorial with Transformers:
# Use a pipeline as a high-level helper
# Warning: Pipeline type "summarization" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline
pipe = pipeline("summarization", model="Hoax0930/summary_tutorial") # Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("Hoax0930/summary_tutorial")
model = AutoModelForSeq2SeqLM.from_pretrained("Hoax0930/summary_tutorial")This model is a fine-tuned version of t5-base on the amazon_reviews_multi 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 |
|---|---|---|---|---|---|---|---|
| 2.867 | 1.0 | 771 | 2.7687 | 18.7278 | 10.8323 | 17.9617 | 17.8262 |
| 2.5363 | 2.0 | 1542 | 2.7741 | 18.4642 | 10.2431 | 17.8719 | 17.7799 |
| 2.3516 | 3.0 | 2313 | 2.7578 | 18.4591 | 9.5235 | 18.0114 | 17.8761 |
| 2.216 | 4.0 | 3084 | 2.7938 | 20.2372 | 10.7222 | 19.5686 | 19.5045 |
| 2.0929 | 5.0 | 3855 | 2.7981 | 19.536 | 10.0232 | 18.7119 | 18.7774 |
| 2.0 | 6.0 | 4626 | 2.8229 | 18.5138 | 9.6349 | 17.7119 | 17.782 |
| 1.9479 | 7.0 | 5397 | 2.8397 | 19.3795 | 10.0475 | 18.5904 | 18.532 |
| 1.9031 | 8.0 | 6168 | 2.8504 | 19.4358 | 10.0475 | 18.6327 | 18.648 |