sentence-transformers
Safetensors
setfit
English
bert
code-security
vulnerability-detection
sast
spidercob
Eval Results (legacy)
Instructions to use Sumeetgpt/code-risk-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use Sumeetgpt/code-risk-classifier with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("Sumeetgpt/code-risk-classifier") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - setfit
How to use Sumeetgpt/code-risk-classifier with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("Sumeetgpt/code-risk-classifier") - Notebooks
- Google Colab
- Kaggle
| { | |
| "base_model": "sentence-transformers/all-MiniLM-L6-v2", | |
| "labels": [ | |
| "REAL_SECRET", | |
| "VULNERABLE_LOGIC", | |
| "TEST_MOCK", | |
| "SAFE_CODE" | |
| ], | |
| "test_accuracy": 0.9808612440191388, | |
| "training_sources": [ | |
| "WebGoat", | |
| "DVWA", | |
| "truffleHog", | |
| "factory_boy", | |
| "faker", | |
| "pytest", | |
| "django", | |
| "fastapi" | |
| ] | |
| } |