Update README.md
Browse files
README.md
CHANGED
@@ -89,7 +89,7 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
89 |
### Training Data
|
90 |
- **Dataset:** `DetectVul/devign`
|
91 |
- **Classes:** `0 (Safe)`, `1 (Vulnerable)`
|
92 |
-
- **Size:**
|
93 |
|
94 |
### Training Procedure
|
95 |
- **Optimizer:** AdamW
|
@@ -102,7 +102,7 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
102 |
|
103 |
### Training Metrics
|
104 |
| Metric | Score |
|
105 |
-
|
106 |
| **Train Loss** | 0.4835 |
|
107 |
| **Evaluation Loss** | 0.6855 |
|
108 |
| **Accuracy** | 68.34% |
|
@@ -125,6 +125,6 @@ The model was evaluated using **20% of the dataset**, with the following results
|
|
125 |
## Environmental Impact
|
126 |
|
127 |
| Factor | Value |
|
128 |
-
|
129 |
| **GPU Used** | 2x T4 GPU |
|
130 |
| **Training Time** | ~1 hour |
|
|
|
89 |
### Training Data
|
90 |
- **Dataset:** `DetectVul/devign`
|
91 |
- **Classes:** `0 (Safe)`, `1 (Vulnerable)`
|
92 |
+
- **Size:** 17483 code snippets
|
93 |
|
94 |
### Training Procedure
|
95 |
- **Optimizer:** AdamW
|
|
|
102 |
|
103 |
### Training Metrics
|
104 |
| Metric | Score |
|
105 |
+
|------------|-------------|
|
106 |
| **Train Loss** | 0.4835 |
|
107 |
| **Evaluation Loss** | 0.6855 |
|
108 |
| **Accuracy** | 68.34% |
|
|
|
125 |
## Environmental Impact
|
126 |
|
127 |
| Factor | Value |
|
128 |
+
|-----------|----------|
|
129 |
| **GPU Used** | 2x T4 GPU |
|
130 |
| **Training Time** | ~1 hour |
|