esdurmus/wiki_lingua
Viewer • Updated • 274k • 1.6k • 56
How to use ahmeddbahaa/arabert2arabert-finetuned-ar-wikilingua 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="ahmeddbahaa/arabert2arabert-finetuned-ar-wikilingua") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("ahmeddbahaa/arabert2arabert-finetuned-ar-wikilingua")
model = AutoModelForSeq2SeqLM.from_pretrained("ahmeddbahaa/arabert2arabert-finetuned-ar-wikilingua", device_map="auto")This model is a fine-tuned version of on the wiki_lingua 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 | Rouge-1 | Rouge-2 | Rouge-l | Gen Len | Bertscore |
|---|---|---|---|---|---|---|---|---|
| 6.7667 | 1.0 | 156 | 5.3846 | 3.36 | 0.56 | 3.27 | 20.0 | 60.6 |
| 5.257 | 2.0 | 312 | 5.0424 | 5.44 | 0.88 | 5.35 | 20.0 | 60.56 |
| 4.743 | 3.0 | 468 | 4.8294 | 9.21 | 1.8 | 8.93 | 20.0 | 62.91 |
| 4.3832 | 4.0 | 624 | 4.7240 | 9.88 | 2.19 | 9.6 | 20.0 | 62.65 |
| 4.1166 | 5.0 | 780 | 4.6861 | 11.61 | 2.86 | 11.13 | 20.0 | 63.71 |
| 3.91 | 6.0 | 936 | 4.6692 | 12.27 | 3.11 | 11.76 | 20.0 | 64.07 |
| 3.7569 | 7.0 | 1092 | 4.6805 | 12.93 | 3.38 | 12.28 | 20.0 | 64.61 |
| 3.6454 | 8.0 | 1248 | 4.6877 | 13.2 | 3.43 | 12.45 | 20.0 | 64.88 |