indonlp/indonlu
Updated • 789 • 41
How to use afbudiman/indobert-distilled-optimized-for-classification with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="afbudiman/indobert-distilled-optimized-for-classification") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("afbudiman/indobert-distilled-optimized-for-classification")
model = AutoModelForSequenceClassification.from_pretrained("afbudiman/indobert-distilled-optimized-for-classification", device_map="auto")This model is a fine-tuned version of distilbert-base-uncased on the indonlu 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 | Accuracy | F1 |
|---|---|---|---|---|---|
| 1.2938 | 1.0 | 688 | 0.8433 | 0.8484 | 0.8513 |
| 0.711 | 2.0 | 1376 | 0.6408 | 0.8881 | 0.8878 |
| 0.4416 | 3.0 | 2064 | 0.7964 | 0.8794 | 0.8793 |
| 0.2907 | 4.0 | 2752 | 0.7559 | 0.8897 | 0.8900 |
| 0.2065 | 5.0 | 3440 | 0.6892 | 0.8968 | 0.8974 |
| 0.1574 | 6.0 | 4128 | 0.6881 | 0.8913 | 0.8906 |
| 0.1131 | 7.0 | 4816 | 0.6224 | 0.8984 | 0.8982 |
| 0.0865 | 8.0 | 5504 | 0.6312 | 0.8976 | 0.8970 |
| 0.0678 | 9.0 | 6192 | 0.6187 | 0.8992 | 0.8989 |
| 0.0526 | 10.0 | 6880 | 0.5991 | 0.9024 | 0.9021 |