ayshajavd/graphcodebert-vuln-classifier at main

graphcodebert-vuln-classifier

338 MB

Ctrl+K

Ctrl+K

1 contributor

History: 6 commits

ayshajavd's picture

Update model card with full test set evaluation metrics (5K samples)

82b783c verified 3 days ago

.gitattributes

1.52 kB
initial commit 3 days ago
README.md

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

1.98 kB
Fix id2label mapping with actual CWE names 3 days ago
label_config.json

5.76 kB
Upload label_config.json with huggingface_hub 3 days ago
model.safetensors

334 MB
xet

GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) 3 days ago
tokenizer.json

3.71 MB
GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) 3 days ago
tokenizer_config.json

406 Bytes
GraphCodeBERT vulnerability classifier (CPU-trained, 20K balanced subset) 3 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) 3 days ago