Update README.md
Browse files
README.md
CHANGED
|
@@ -101,7 +101,7 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
| 101 |
- **Hardware Used:** 2x T4 GPU
|
| 102 |
- **Mixed Precision:** FP16
|
| 103 |
|
| 104 |
-
###
|
| 105 |
| Metric | Score |
|
| 106 |
|------------|-------------|
|
| 107 |
| **Train Loss** | 0.4835 |
|
|
@@ -111,18 +111,6 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
| 111 |
| **Precision** | 69.18% |
|
| 112 |
| **Recall** | 56.40% |
|
| 113 |
|
| 114 |
-
## Evaluation
|
| 115 |
-
|
| 116 |
-
### Testing Data & Metrics
|
| 117 |
-
The model was evaluated using **20% of the dataset**, with the following results:
|
| 118 |
-
|
| 119 |
-
- **Evaluation Accuracy:** 68.34%
|
| 120 |
-
- **F1 Score:** 62.14%
|
| 121 |
-
- **Precision:** 69.18%
|
| 122 |
-
- **Recall:** 56.40%
|
| 123 |
-
- **Evaluation Runtime:** 41.16 sec
|
| 124 |
-
- **Evaluation Speed:** 53.1 samples/sec
|
| 125 |
-
|
| 126 |
## Environmental Impact
|
| 127 |
|
| 128 |
| Factor | Value |
|
|
|
|
| 101 |
- **Hardware Used:** 2x T4 GPU
|
| 102 |
- **Mixed Precision:** FP16
|
| 103 |
|
| 104 |
+
### Metrics
|
| 105 |
| Metric | Score |
|
| 106 |
|------------|-------------|
|
| 107 |
| **Train Loss** | 0.4835 |
|
|
|
|
| 111 |
| **Precision** | 69.18% |
|
| 112 |
| **Recall** | 56.40% |
|
| 113 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 114 |
## Environmental Impact
|
| 115 |
|
| 116 |
| Factor | Value |
|