Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

ayshajavd
/
graphcodebert-vuln-classifier

Text Classification
Transformers
Safetensors
English
code
roberta
security
vulnerability-detection
code-analysis
multi-label-classification
graphcodebert
owasp
cwe
static-analysis
Eval Results (legacy)
text-embeddings-inference
Model card Files Files and versions
xet
Community

Instructions to use ayshajavd/graphcodebert-vuln-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use ayshajavd/graphcodebert-vuln-classifier with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="ayshajavd/graphcodebert-vuln-classifier")
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForSequenceClassification
    
    tokenizer = AutoTokenizer.from_pretrained("ayshajavd/graphcodebert-vuln-classifier")
    model = AutoModelForSequenceClassification.from_pretrained("ayshajavd/graphcodebert-vuln-classifier")
  • Notebooks
  • Google Colab
  • Kaggle
graphcodebert-vuln-classifier
502 MB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 7 commits
ayshajavd's picture
ayshajavd
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration
b329c79 verified 9 days ago
  • .gitattributes
    1.52 kB
    initial commit 14 days ago
  • README.md
    9.58 kB
    Update model card with full test set evaluation metrics (5K samples) 14 days ago
  • config.json
    2.1 kB
    v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 9 days ago
  • label_config.json
    12.6 kB
    v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 9 days ago
  • model.safetensors
    499 MB
    xet
    v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 9 days ago
  • tokenizer.json
    3.56 MB
    v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 9 days ago
  • tokenizer_config.json
    359 Bytes
    v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 9 days ago
  • training_args.bin
    5.33 kB
    xet
    GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) 14 days ago