metadata
library_name: transformers
license: apache-2.0
base_model: distilbert-base-uncased
tags:
- generated_from_trainer
model-index:
- name: junior_agent
results: []
junior_agent
This model is a fine-tuned version of distilbert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 0.5637
- Topology Accuracy: 0.8431
- Service Accuracy: 0.9082
- Combined Accuracy: 0.8757
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: 5e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- lr_scheduler_warmup_steps: 50
- num_epochs: 15
- mixed_precision_training: Native AMP
Training results
| Training Loss | Epoch | Step | Validation Loss | Topology Accuracy | Service Accuracy | Combined Accuracy |
|---|---|---|---|---|---|---|
| 0.8918 | 1.0 | 143 | 0.8370 | 0.7367 | 0.5824 | 0.6596 |
| 0.7634 | 2.0 | 286 | 0.6906 | 0.7872 | 0.8045 | 0.7959 |
| 0.6648 | 3.0 | 429 | 0.5570 | 0.8351 | 0.8896 | 0.8624 |
| 0.4889 | 4.0 | 572 | 0.5268 | 0.8351 | 0.9109 | 0.8730 |
| 0.4557 | 5.0 | 715 | 0.5296 | 0.8378 | 0.9109 | 0.8743 |
| 0.4257 | 6.0 | 858 | 0.5382 | 0.8364 | 0.9189 | 0.8777 |
| 0.4398 | 7.0 | 1001 | 0.5383 | 0.8497 | 0.9215 | 0.8856 |
| 0.3671 | 8.0 | 1144 | 0.5535 | 0.8497 | 0.9176 | 0.8836 |
| 0.3781 | 9.0 | 1287 | 0.5637 | 0.8431 | 0.9082 | 0.8757 |
Framework versions
- Transformers 4.56.1
- Pytorch 2.8.0+cu126
- Datasets 4.0.0
- Tokenizers 0.22.0