Update README.md
Browse files
README.md
CHANGED
|
@@ -94,6 +94,7 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
| 94 |
### Training Data
|
| 95 |
- **Dataset:** `DetectVul/devign`
|
| 96 |
- **Classes:** `0 (Safe)`, `1 (Vulnerable)`
|
|
|
|
| 97 |
|
| 98 |
### Training Procedure
|
| 99 |
- **Optimizer:** AdamW
|
|
@@ -101,7 +102,7 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
| 101 |
- **Batch Size:** 8
|
| 102 |
- **Learning Rate:** 2e-05
|
| 103 |
- **Epochs:** 3
|
| 104 |
-
- **Hardware Used:** T4 GPU
|
| 105 |
|
| 106 |
### Metrics
|
| 107 |
| Metric | Score |
|
|
|
|
| 94 |
### Training Data
|
| 95 |
- **Dataset:** `DetectVul/devign`
|
| 96 |
- **Classes:** `0 (Safe)`, `1 (Vulnerable)`
|
| 97 |
+
- **Size:** `21800` Code Snippets
|
| 98 |
|
| 99 |
### Training Procedure
|
| 100 |
- **Optimizer:** AdamW
|
|
|
|
| 102 |
- **Batch Size:** 8
|
| 103 |
- **Learning Rate:** 2e-05
|
| 104 |
- **Epochs:** 3
|
| 105 |
+
- **Hardware Used:** 2x T4 GPU
|
| 106 |
|
| 107 |
### Metrics
|
| 108 |
| Metric | Score |
|