Instructions to use mahdin70/unixcoder-code-vulnerability-detector with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mahdin70/unixcoder-code-vulnerability-detector with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mahdin70/unixcoder-code-vulnerability-detector")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mahdin70/unixcoder-code-vulnerability-detector") model = AutoModelForSequenceClassification.from_pretrained("mahdin70/unixcoder-code-vulnerability-detector") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -99,7 +99,6 @@ print("Vulnerable Code" if predicted_label == 1 else "Safe Code")
|
|
| 99 |
- **Learning Rate:** 2e-5
|
| 100 |
- **Epochs:** 3
|
| 101 |
- **Hardware Used:** 2x T4 GPU
|
| 102 |
-
- **Mixed Precision:** FP32
|
| 103 |
|
| 104 |
### Metrics
|
| 105 |
| Metric | Score |
|
|
|
|
| 99 |
- **Learning Rate:** 2e-5
|
| 100 |
- **Epochs:** 3
|
| 101 |
- **Hardware Used:** 2x T4 GPU
|
|
|
|
| 102 |
|
| 103 |
### Metrics
|
| 104 |
| Metric | Score |
|