Khansa-saAI-29's picture
Update README.md
3a937a1 verified
|
Raw
History Blame Contribute Delete
740 Bytes
---
license: apache-2.0
language:
- code
library_name: transformers
pipeline_tag: text-classification
base_model: microsoft/codebert-base
datasets:
- DiverseVul
tags:
- codebert
- vulnerability-detection
- cybersecurity
- software-security
- static-analysis
- transformers
metrics:
- accuracy
- precision
- recall
- f1
---
# Fine-tuned CodeBERT for Vulnerability Detection---
license: apache-2.0
base_model:
- microsoft/codebert-base
library_name: transformers
---
# Fine-tuned CodeBERT for Vulnerability Detection
## Description
This model is fine-tuned on the Big-Vul dataset for binary vulnerability classification.
Labels:
- 0 = Non-Vulnerable
- 1 = Vulnerable
Base model:
microsoft/codebert-base
Framework:
Transformers + PyTorch