Model: mulliken/cwe-predictor
Author: mulliken
Pipeline: text-classification
Library: transformers
Tags: transformers, safetensors, distilbert, text-classification, security, cyber-security, CWE, vulnerability-classification, cve, en, dataset:zefang-liu/cve-and-cwe-mapping-dataset, arxiv:1910.09700, base_model:distilbert/distilbert-base-uncased, base_model:finetune:distilbert/distilbert-base-uncased, license:apache-2.0, model-index, endpoints_compatible, region:us
Base models (finetune): id: distilbert/distilbert-base-uncased
Card metadata: library_name: transformers
tags: security, cyber-security, CWE, vulnerability-classification, cve
license: apache-2.0
datasets: zefang-liu/cve-and-cwe-mapping-dataset
language: en
metrics: accuracy, f1
base_model: distilbert/distilbert-base-uncased
pipeline_tag: text-classification
model-index: name: cwe-predictor
results: task: type: text-classification
name: CWE Classification
metrics: type: accuracy
value: 0.727207
name: Validation Accuracy
verified: no, type: f1
value: 0.251264
name: Macro F1 Score
verified: no
Likes: 0 | All-time downloads: 137
Created: 2025-08-12 12:12:59