Text Classification
Transformers
Safetensors
code
cybersecurity
vulnerability
cpp
Mohammed Sbaihi commited on
Commit
4e46543
·
verified ·
1 Parent(s): 28743a5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -20,7 +20,7 @@ pipeline_tag: text-classification
20
  <!-- ![deck](deck.png){: 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. Instead of binary classification ("safe", "unsafe"), it classify the input code into 7 labels: 'safe' (no vulnerability detected) and six other CWE weaknesses:
 
 
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
  <!-- ![deck](deck.png){: 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