klue/klue
Viewer • Updated • 206k • 3.41k • 96
How to use ys7yoo/sts_klue_roberta_large_ep9 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="ys7yoo/sts_klue_roberta_large_ep9") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ys7yoo/sts_klue_roberta_large_ep9")
model = AutoModelForSequenceClassification.from_pretrained("ys7yoo/sts_klue_roberta_large_ep9", device_map="auto")This model is a fine-tuned version of klue/roberta-large on the klue 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 | Mse | Mae | R2 |
|---|---|---|---|---|---|---|
| 1.3093 | 1.0 | 183 | 0.4915 | 0.4915 | 0.5401 | 0.7750 |
| 0.2188 | 2.0 | 366 | 0.4399 | 0.4399 | 0.4982 | 0.7986 |
| 0.1327 | 3.0 | 549 | 0.4022 | 0.4022 | 0.4647 | 0.8158 |
| 0.1043 | 4.0 | 732 | 0.4094 | 0.4094 | 0.4680 | 0.8125 |
| 0.074 | 5.0 | 915 | 0.4218 | 0.4218 | 0.4784 | 0.8069 |
| 0.0552 | 6.0 | 1098 | 0.3424 | 0.3424 | 0.4356 | 0.8432 |
| 0.0394 | 7.0 | 1281 | 0.3925 | 0.3925 | 0.4691 | 0.8203 |
| 0.031 | 8.0 | 1464 | 0.3723 | 0.3723 | 0.4510 | 0.8295 |
| 0.0234 | 9.0 | 1647 | 0.3567 | 0.3567 | 0.4407 | 0.8367 |
Base model
klue/roberta-large