ayshajavd/graphcodebert-vuln-classifier at main

graphcodebert-vuln-classifier

502 MB

Ctrl+K

Ctrl+K

1 contributor

History: 7 commits

ayshajavd's picture

v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration

b329c79 verified about 15 hours ago

.gitattributes

1.52 kB
initial commit 5 days ago
README.md

9.58 kB
Update model card with full test set evaluation metrics (5K samples) 5 days ago
config.json

2.1 kB
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration about 15 hours ago
label_config.json

12.6 kB
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration about 15 hours ago
model.safetensors

499 MB
xet

v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration about 15 hours ago
tokenizer.json

3.56 MB
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration about 15 hours ago
tokenizer_config.json

359 Bytes
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration about 15 hours ago
training_args.bin
Detected Pickle imports (10)
- "transformers.trainer_utils.SaveStrategy",
- "transformers.training_args.OptimizerNames",
- "transformers.trainer_utils.SchedulerType",
- "transformers.trainer_utils.HubStrategy",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "transformers.training_args.TrainingArguments",
- "torch.device",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.trainer_utils.IntervalStrategy",
- "accelerate.state.PartialState"
How to fix it?
5.33 kB
xet

GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) 5 days ago