fancyzhx/ag_news
Viewer • Updated • 128k • 111k • 190
How to use TransferGraph/ChrisUPM_BioBERT_Re_trained-finetuned-lora-ag_news with PEFT:
from peft import PeftModel
from transformers import AutoModelForSequenceClassification
base_model = AutoModelForSequenceClassification.from_pretrained("ChrisUPM/BioBERT_Re_trained")
model = PeftModel.from_pretrained(base_model, "TransferGraph/ChrisUPM_BioBERT_Re_trained-finetuned-lora-ag_news")This model is a fine-tuned version of ChrisUPM/BioBERT_Re_trained on the ag_news 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:
| accuracy | train_loss | epoch |
|---|---|---|
| 0.2166 | None | 0 |
| 0.8895 | 0.4140 | 0 |
| 0.8949 | 0.3025 | 1 |
| 0.9034 | 0.2748 | 2 |
| 0.9046 | 0.2582 | 3 |
Base model
ChrisUPM/BioBERT_Re_trained