Hugging Face's logo

mcanoglu
/
deepseek-ai-deepseek-coder-1.3b-base-finetuned-defect-cwe-group-detection

Text Classification
Transformers
TensorBoard
Safetensors
llama
Generated from Trainer
text-generation-inference
Inference Endpoints
Model card Files Metrics Community
1
mcanoglu commited on
Commit
c8b0bc4
•
1 Parent(s): fbc4405

End of training

Browse files
Files changed (3) hide show
  1. README.md +9 -9
  2. model-00001-of-00002.safetensors +1 -1
  3. model-00002-of-00002.safetensors +1 -1
README.md CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [deepseek-ai/deepseek-coder-1.3b-base](https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-base) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.5587
24
- - Accuracy: 0.7659
25
- - F1: 0.7659
26
- - Precision: 0.7659
27
- - Recall: 0.7659
28
 
29
  ## Model description
30
 
@@ -58,9 +58,9 @@ The following hyperparameters were used during training:
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
60
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
61
- | No log | 1.0 | 462 | 0.4717 | 0.7772 | 0.7772 | 0.7772 | 0.7772 |
62
- | 0.5768 | 2.0 | 924 | 0.4683 | 0.7681 | 0.7681 | 0.7681 | 0.7681 |
63
- | 0.3029 | 3.0 | 1386 | 0.5587 | 0.7659 | 0.7659 | 0.7659 | 0.7659 |
64
 
65
 
66
  ### Framework versions
@@ -68,4 +68,4 @@ The following hyperparameters were used during training:
68
  - Transformers 4.37.0
69
  - Pytorch 2.1.2+cu121
70
  - Datasets 2.16.1
71
- - Tokenizers 0.15.0
 
20
 
21
  This model is a fine-tuned version of [deepseek-ai/deepseek-coder-1.3b-base](https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-base) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.5777
24
+ - Accuracy: 0.7586
25
+ - F1: 0.7499
26
+ - Precision: 0.7513
27
+ - Recall: 0.7586
28
 
29
  ## Model description
30
 
 
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
60
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
61
+ | No log | 1.0 | 462 | 0.4832 | 0.7743 | 0.7594 | 0.7720 | 0.7743 |
62
+ | 0.5829 | 2.0 | 924 | 0.4705 | 0.7788 | 0.7700 | 0.7737 | 0.7788 |
63
+ | 0.3078 | 3.0 | 1386 | 0.5777 | 0.7586 | 0.7499 | 0.7513 | 0.7586 |
64
 
65
 
66
  ### Framework versions
 
68
  - Transformers 4.37.0
69
  - Pytorch 2.1.2+cu121
70
  - Datasets 2.16.1
71
+ - Tokenizers 0.15.1
model-00001-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e6692dc047c116f58c8b10194bb1595573450029296bf03f683110ef6ef87a98
3
  size 4986380064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a209cfe6427ea97e774bc64ad0d1a2e36d201e528ad7a3957d568d7aad544fc5
3
  size 4986380064
model-00002-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:245f6e59e33b617159a97cb1a4a1571e1b5a7362fcda1e3ad14a4c29dcbd4131
3
  size 135332592
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37a2bfdc5d4bac777ff692b2a728046862284a7f87ae73e9455be85302563498
3
  size 135332592