Model save
Browse files- README.md +24 -48
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -16,19 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 16 |
|
| 17 |
This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on the None dataset.
|
| 18 |
It achieves the following results on the evaluation set:
|
| 19 |
-
- Loss: 0.
|
| 20 |
-
- Complexity Accuracy: 0.
|
| 21 |
-
- Model Accuracy: 0.
|
| 22 |
-
- Overall Accuracy: 0.
|
| 23 |
-
- Comp Acc Class 0: 0.9333
|
| 24 |
-
- Comp Acc Class 1: 0.7614
|
| 25 |
-
- Comp Acc Class 2: 0.7030
|
| 26 |
-
- Model Acc Class 0: 0.4166
|
| 27 |
-
- Model Acc Class 1: 0.3263
|
| 28 |
-
- Model Acc Class 2: 0.3706
|
| 29 |
-
- Model Acc Class 3: 0.224
|
| 30 |
-
- Complexity Macro F1: 0.7794
|
| 31 |
-
- Model Macro F1: 0.2693
|
| 32 |
|
| 33 |
## Model description
|
| 34 |
|
|
@@ -47,50 +38,35 @@ More information needed
|
|
| 47 |
### Training hyperparameters
|
| 48 |
|
| 49 |
The following hyperparameters were used during training:
|
| 50 |
-
- learning_rate: 0.
|
| 51 |
-
- train_batch_size:
|
| 52 |
- eval_batch_size: 32
|
| 53 |
- seed: 42
|
| 54 |
- gradient_accumulation_steps: 2
|
| 55 |
-
- total_train_batch_size:
|
| 56 |
- optimizer: Use adamw_torch_fused with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
| 57 |
- lr_scheduler_type: cosine
|
| 58 |
-
- lr_scheduler_warmup_ratio: 0.
|
| 59 |
- num_epochs: 10
|
| 60 |
|
| 61 |
### Training results
|
| 62 |
|
| 63 |
-
| Training Loss | Epoch | Step | Validation Loss | Complexity Accuracy | Model Accuracy | Overall Accuracy |
|
| 64 |
-
|
| 65 |
-
| 0.
|
| 66 |
-
| 0.
|
| 67 |
-
| 0.
|
| 68 |
-
| 0.
|
| 69 |
-
| 0.
|
| 70 |
-
| 0.
|
| 71 |
-
| 0.
|
| 72 |
-
| 0.
|
| 73 |
-
| 0.
|
| 74 |
-
| 0.
|
| 75 |
-
| 0.
|
| 76 |
-
| 0.
|
| 77 |
-
| 0.
|
| 78 |
-
| 0.
|
| 79 |
-
| 0.6826 | 5.1429 | 4500 | 0.8959 | 0.7626 | 0.3900 | 0.2854 | 0.9147 | 0.7535 | 0.6980 | 0.4102 | 0.2649 | 0.4328 | 0.296 | 0.7723 | 0.2708 |
|
| 80 |
-
| 0.6957 | 5.4857 | 4800 | 0.8982 | 0.7719 | 0.4095 | 0.3014 | 0.8915 | 0.8049 | 0.6493 | 0.4439 | 0.2399 | 0.3930 | 0.352 | 0.7752 | 0.2771 |
|
| 81 |
-
| 0.6686 | 5.8286 | 5100 | 0.8987 | 0.7631 | 0.4095 | 0.2939 | 0.9271 | 0.7437 | 0.7104 | 0.4493 | 0.2284 | 0.3856 | 0.296 | 0.7732 | 0.2716 |
|
| 82 |
-
| 0.6681 | 6.1714 | 5400 | 0.8870 | 0.7694 | 0.4147 | 0.3046 | 0.9566 | 0.7558 | 0.6939 | 0.4580 | 0.2860 | 0.2960 | 0.304 | 0.7789 | 0.2731 |
|
| 83 |
-
| 0.6526 | 6.5143 | 5700 | 0.8983 | 0.7746 | 0.3980 | 0.2892 | 0.9426 | 0.7692 | 0.6947 | 0.4243 | 0.2726 | 0.4055 | 0.272 | 0.7815 | 0.2714 |
|
| 84 |
-
| 0.656 | 6.8571 | 6000 | 0.9080 | 0.7582 | 0.4 | 0.2889 | 0.9364 | 0.7164 | 0.7376 | 0.4216 | 0.2879 | 0.4527 | 0.184 | 0.7712 | 0.2708 |
|
| 85 |
-
| 0.6421 | 7.2 | 6300 | 0.9028 | 0.7768 | 0.3893 | 0.2889 | 0.9349 | 0.8072 | 0.6386 | 0.4068 | 0.3551 | 0.3507 | 0.24 | 0.7791 | 0.2690 |
|
| 86 |
-
| 0.6366 | 7.5429 | 6600 | 0.8946 | 0.7731 | 0.4087 | 0.3019 | 0.9504 | 0.7905 | 0.6477 | 0.4425 | 0.2399 | 0.4254 | 0.256 | 0.7764 | 0.2729 |
|
| 87 |
-
| 0.6463 | 7.8857 | 6900 | 0.8931 | 0.7706 | 0.4092 | 0.2986 | 0.9380 | 0.7581 | 0.7038 | 0.4419 | 0.3186 | 0.3358 | 0.248 | 0.7790 | 0.2740 |
|
| 88 |
-
| 0.6405 | 8.2286 | 7200 | 0.8919 | 0.7706 | 0.4110 | 0.3009 | 0.9380 | 0.7581 | 0.7038 | 0.4415 | 0.2802 | 0.4129 | 0.224 | 0.7784 | 0.2756 |
|
| 89 |
-
| 0.6101 | 8.5714 | 7500 | 0.9054 | 0.7729 | 0.3988 | 0.2939 | 0.9302 | 0.7614 | 0.7096 | 0.4247 | 0.3225 | 0.3607 | 0.224 | 0.7813 | 0.2700 |
|
| 90 |
-
| 0.6366 | 8.9143 | 7800 | 0.9038 | 0.7696 | 0.3995 | 0.2932 | 0.9240 | 0.7567 | 0.7104 | 0.4250 | 0.3186 | 0.3706 | 0.224 | 0.7778 | 0.2711 |
|
| 91 |
-
| 0.6306 | 9.2571 | 8100 | 0.9099 | 0.7704 | 0.3915 | 0.2859 | 0.9240 | 0.7521 | 0.7211 | 0.4125 | 0.3186 | 0.3856 | 0.216 | 0.7794 | 0.2682 |
|
| 92 |
-
| 0.6293 | 9.6 | 8400 | 0.9093 | 0.7716 | 0.3938 | 0.2889 | 0.9333 | 0.7618 | 0.7030 | 0.4159 | 0.3244 | 0.3731 | 0.224 | 0.7795 | 0.2691 |
|
| 93 |
-
| 0.6177 | 9.9429 | 8700 | 0.9091 | 0.7714 | 0.3943 | 0.2892 | 0.9333 | 0.7614 | 0.7030 | 0.4166 | 0.3263 | 0.3706 | 0.224 | 0.7794 | 0.2693 |
|
| 94 |
|
| 95 |
|
| 96 |
### Framework versions
|
|
|
|
| 16 |
|
| 17 |
This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on the None dataset.
|
| 18 |
It achieves the following results on the evaluation set:
|
| 19 |
+
- Loss: 0.6852
|
| 20 |
+
- Complexity Accuracy: 0.772
|
| 21 |
+
- Model Accuracy: 0.747
|
| 22 |
+
- Overall Accuracy: 0.5793
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
## Model description
|
| 25 |
|
|
|
|
| 38 |
### Training hyperparameters
|
| 39 |
|
| 40 |
The following hyperparameters were used during training:
|
| 41 |
+
- learning_rate: 0.0001
|
| 42 |
+
- train_batch_size: 32
|
| 43 |
- eval_batch_size: 32
|
| 44 |
- seed: 42
|
| 45 |
- gradient_accumulation_steps: 2
|
| 46 |
+
- total_train_batch_size: 64
|
| 47 |
- optimizer: Use adamw_torch_fused with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
| 48 |
- lr_scheduler_type: cosine
|
| 49 |
+
- lr_scheduler_warmup_ratio: 0.01
|
| 50 |
- num_epochs: 10
|
| 51 |
|
| 52 |
### Training results
|
| 53 |
|
| 54 |
+
| Training Loss | Epoch | Step | Validation Loss | Complexity Accuracy | Model Accuracy | Overall Accuracy |
|
| 55 |
+
|:-------------:|:------:|:----:|:---------------:|:-------------------:|:--------------:|:----------------:|
|
| 56 |
+
| 0.8284 | 0.6857 | 300 | 0.7391 | 0.7275 | 0.7475 | 0.5437 |
|
| 57 |
+
| 0.7657 | 1.3703 | 600 | 0.7173 | 0.7408 | 0.7478 | 0.5515 |
|
| 58 |
+
| 0.7398 | 2.0549 | 900 | 0.7099 | 0.7502 | 0.748 | 0.5595 |
|
| 59 |
+
| 0.7161 | 2.7406 | 1200 | 0.7037 | 0.7578 | 0.748 | 0.5645 |
|
| 60 |
+
| 0.7057 | 3.4251 | 1500 | 0.6973 | 0.7635 | 0.7468 | 0.569 |
|
| 61 |
+
| 0.7115 | 4.1097 | 1800 | 0.6927 | 0.764 | 0.748 | 0.5705 |
|
| 62 |
+
| 0.7214 | 4.7954 | 2100 | 0.6896 | 0.7672 | 0.7482 | 0.5755 |
|
| 63 |
+
| 0.7034 | 5.48 | 2400 | 0.6886 | 0.769 | 0.7472 | 0.5777 |
|
| 64 |
+
| 0.6935 | 6.1646 | 2700 | 0.6878 | 0.769 | 0.7478 | 0.577 |
|
| 65 |
+
| 0.7055 | 6.8503 | 3000 | 0.6867 | 0.7722 | 0.7465 | 0.5787 |
|
| 66 |
+
| 0.6983 | 7.5349 | 3300 | 0.6858 | 0.7728 | 0.7465 | 0.5797 |
|
| 67 |
+
| 0.7092 | 8.2194 | 3600 | 0.6849 | 0.774 | 0.747 | 0.5803 |
|
| 68 |
+
| 0.697 | 8.9051 | 3900 | 0.6851 | 0.7718 | 0.747 | 0.5787 |
|
| 69 |
+
| 0.6989 | 9.5897 | 4200 | 0.6852 | 0.772 | 0.747 | 0.5793 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
### Framework versions
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 597632156
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ddda65714e2f58101eb9cce10d5d5bdc15236e1a9f7cdf53455428d7a54cde39
|
| 3 |
size 597632156
|