Model save

README.md

@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [huggingface/CodeBERTa-small-v1](https://huggingface.co/huggingface/CodeBERTa-small-v1) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0041
-- F1: 1.0
-- Accuracy: 1.0
-- Precision: 1.0
-- Recall: 1.0
+- Loss: 0.0171
+- F1: 0.9975
+- Accuracy: 0.9975
+- Precision: 0.9975
+- Recall: 0.9975
 
 ## Model description
 
@@ -50,15 +50,20 @@ The following hyperparameters were used during training:
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
-- training_steps: 100
+- num_epochs: 1
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | F1     | Accuracy | Precision | Recall |
-|:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:---------:|:------:|
-| No log        | 0     | 0    | 0.6985          | 0.3223 | 0.49     | 0.2401    | 0.49   |
-| 0.0001        | 12.5  | 50   | 0.0044          | 1.0    | 1.0      | 1.0       | 1.0    |
-| 0.0001        | 25.0  | 100  | 0.0041          | 1.0    | 1.0      | 1.0       | 1.0    |
+| Training Loss | Epoch  | Step | Validation Loss | F1     | Accuracy | Precision | Recall |
+|:-------------:|:------:|:----:|:---------------:|:------:|:--------:|:---------:|:------:|
+| No log        | 0      | 0    | 0.6981          | 0.3337 | 0.5001   | 0.6162    | 0.5001 |
+| 0.0294        | 0.1420 | 1000 | 0.0398          | 0.9947 | 0.9947   | 0.9947    | 0.9947 |
+| 0.0076        | 0.2841 | 2000 | 0.0211          | 0.9968 | 0.9968   | 0.9968    | 0.9968 |
+| 0.0053        | 0.4261 | 3000 | 0.0188          | 0.9973 | 0.9973   | 0.9973    | 0.9973 |
+| 0.0056        | 0.5681 | 4000 | 0.0166          | 0.9976 | 0.9976   | 0.9976    | 0.9976 |
+| 0.0044        | 0.7101 | 5000 | 0.0172          | 0.9975 | 0.9975   | 0.9975    | 0.9975 |
+| 0.0009        | 0.8522 | 6000 | 0.0171          | 0.9975 | 0.9975   | 0.9975    | 0.9975 |
+| 0.0052        | 0.9942 | 7000 | 0.0171          | 0.9975 | 0.9975   | 0.9975    | 0.9975 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7ed1b6ac966d26401991563486e96ffeb0db29524abdd2359e021e9f7935639c
+oid sha256:8faf588219b1a6f558bfe2063e6be08580cb5701573286896a908d8e691f9ddf
 size 166917412