Commit
·
f7a3905
1
Parent(s):
c2cbb97
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,6 @@ I finetuned a RobertaForSequenceClassification model which is initialized
|
|
| 18 |
from CodeBert [https://huggingface.co/microsoft/codebert-base] to judge whether a code is vulnerable or not.
|
| 19 |
I selected balanced samples from MSR dataset [https://github.com/ZeoVan/MSR_20_Code_vulnerability_CSV_Dataset] for training, validation, and testing.
|
| 20 |
The "func_before" is used for code classification. All the data is in the file "msr.csv".
|
| 21 |
-
Funcs shorter than 50 or longer than 512 (The CodeBert window size) are dropped.
|
| 22 |
|
| 23 |
Test Reulsts:
|
| 24 |
acc 0.7022935779816514, f1 0.6482384823848238, precision 0.7920529801324503, recall 0.5486238532110091
|
|
|
|
| 18 |
from CodeBert [https://huggingface.co/microsoft/codebert-base] to judge whether a code is vulnerable or not.
|
| 19 |
I selected balanced samples from MSR dataset [https://github.com/ZeoVan/MSR_20_Code_vulnerability_CSV_Dataset] for training, validation, and testing.
|
| 20 |
The "func_before" is used for code classification. All the data is in the file "msr.csv".
|
|
|
|
| 21 |
|
| 22 |
Test Reulsts:
|
| 23 |
acc 0.7022935779816514, f1 0.6482384823848238, precision 0.7920529801324503, recall 0.5486238532110091
|