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 |
|