Instructions to use Mikelezbe/doku-bert-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Mikelezbe/doku-bert-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Mikelezbe/doku-bert-base")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Mikelezbe/doku-bert-base") model = AutoModelForSequenceClassification.from_pretrained("Mikelezbe/doku-bert-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
doku-bert-base
This model is a fine-tuned version of google-bert/bert-base-cased on the None dataset. It achieves the following results on the evaluation set:
- Loss: 3.1640
- Micro F1: 0.2702
- Precision: 0.2702
- Recall: 0.2702
- F1 Class 0: 0.3052
- F1 Class 1: 0.0
- F1 Class 2: 0.0
- F1 Class 3: 0.2892
- F1 Class 4: 0.0
- F1 Class 5: 0.3551
- F1 Class 6: 0.0
- F1 Class 7: 0.2
- F1 Class 8: 0.2136
- F1 Class 9: 0.325
- F1 Class 10: 0.1461
- F1 Class 11: 0.125
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 1
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 10
Training results
| Training Loss | Epoch | Step | Validation Loss | Micro F1 | Precision | Recall | F1 Class 0 | F1 Class 1 | F1 Class 2 | F1 Class 3 | F1 Class 4 | F1 Class 5 | F1 Class 6 | F1 Class 7 |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 1.0003 | 1.0 | 773 | 0.9631 | 0.6708 | 0.6708 | 0.6708 | 0.6598 | 0.4854 | 0.8137 | 0.6197 | 0.6780 | 0.7798 | 0.7552 | 0.6065 |
| 0.7081 | 2.0 | 1546 | 0.9651 | 0.6753 | 0.6753 | 0.6753 | 0.6404 | 0.5292 | 0.8264 | 0.6331 | 0.6739 | 0.7844 | 0.7443 | 0.6428 |
| 0.4368 | 3.0 | 2319 | 1.1076 | 0.6747 | 0.6747 | 0.6747 | 0.6401 | 0.5432 | 0.8299 | 0.6453 | 0.6641 | 0.7619 | 0.7471 | 0.6304 |
| 0.2398 | 4.0 | 3092 | 1.4374 | 0.6572 | 0.6572 | 0.6572 | 0.6217 | 0.4633 | 0.8248 | 0.6231 | 0.6638 | 0.7380 | 0.7084 | 0.6237 |
| 0.1427 | 5.0 | 3865 | 1.6852 | 0.6611 | 0.6611 | 0.6611 | 0.6352 | 0.5165 | 0.8333 | 0.6317 | 0.6423 | 0.7503 | 0.7258 | 0.6340 |
Framework versions
- Transformers 4.47.0
- Pytorch 2.5.1+cu121
- Datasets 3.1.0
- Tokenizers 0.21.0
- Downloads last month
- 3
Model tree for Mikelezbe/doku-bert-base
Base model
google-bert/bert-base-cased