starmage520 commited on
Commit
f7a3905
1 Parent(s): c2cbb97

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -1
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