Mohammed Sbaihi
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -20,7 +20,7 @@ pipeline_tag: text-classification
|
|
20 |
<!-- {: width="200px"} -->
|
21 |
<img src="deck.png" width="800">
|
22 |
|
23 |
-
This is derivative version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base). <br>
|
24 |
We fine-tuned ModernBERT-base to detect vulnerability in C/C++ Code. <br>
|
25 |
The actual version has an accuracy of 82% <br>
|
26 |
|
@@ -28,7 +28,9 @@ The actual version has an accuracy of 82% <br>
|
|
28 |
|
29 |
### Model Description
|
30 |
|
31 |
-
ThreatDetect-C-Cpp can be used as a code classifier.
|
|
|
|
|
32 |
| Label | Description |
|
33 |
|---------|-------------------------------------------------------|
|
34 |
| CWE-119 | Improper Restriction of Operations within the Bounds of a Memory Buffer |
|
@@ -105,7 +107,7 @@ The model was trained using LoRA applied to Q and V matrices.
|
|
105 |
|
106 |
## Evaluation
|
107 |
|
108 |
-
ThreatDetect-C-Cpp reaches an accruacy of 82
|
109 |
|
110 |
|
111 |
|
|
|
20 |
<!-- {: width="200px"} -->
|
21 |
<img src="deck.png" width="800">
|
22 |
|
23 |
+
This is a derivative version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base). <br>
|
24 |
We fine-tuned ModernBERT-base to detect vulnerability in C/C++ Code. <br>
|
25 |
The actual version has an accuracy of 82% <br>
|
26 |
|
|
|
28 |
|
29 |
### Model Description
|
30 |
|
31 |
+
ThreatDetect-C-Cpp can be used as a code classifier. <br>
|
32 |
+
Instead of binary classification ("safe", "unsafe"), The model classifies the input code into 7 labels: 'safe' (no vulnerability detected) and six other CWE weaknesses:
|
33 |
+
|
34 |
| Label | Description |
|
35 |
|---------|-------------------------------------------------------|
|
36 |
| CWE-119 | Improper Restriction of Operations within the Bounds of a Memory Buffer |
|
|
|
107 |
|
108 |
## Evaluation
|
109 |
|
110 |
+
ThreatDetect-C-Cpp reaches an accruacy of 82% on the eval set.
|
111 |
|
112 |
|
113 |
|