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 6 days ago

.gitattributes

1.52 kB
initial commit 11 days ago
README.md

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

2.1 kB
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 6 days ago
label_config.json

12.6 kB
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 6 days ago
model.safetensors

499 MB
xet

v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 6 days ago
tokenizer.json

3.56 MB
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 6 days ago
tokenizer_config.json

359 Bytes
v2: GraphCodeBERT + ASL + two-phase + per-class thresholds + calibration 6 days 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) 10 days ago