Downloads · 30 days
10
11% of all-time downloads
ARUNAGIRINATHAN/vuln-ner-bert
vuln-ner-bert is a token classification model from ARUNAGIRINATHAN. Use it when you need labels on individual words, such as names. The card lists the license as apache-2.0.
Task: token-classification (NER) for security vulnerability reports. Tags: token-classification, ner, security, cve, cvss
Downloads · 30 days
10
11% of all-time downloads
All-time downloads
93
Public
Parameters
109M
436 MB on disk
Likes
3
Public
Click a slice to open those files.
.safetensors436 MB · 100%
From the Hugging Face model README
Task: token-classification (NER) for security vulnerability reports. Tags: token-classification, ner, security, cve, cvss
from transformers import BertTokenizerFast, BertForTokenClassification
tok = BertTokenizerFast.from_pretrained("ARUNAGIRINATHAN/vuln-ner-bert")
model = BertForTokenClassification.from_pretrained("ARUNAGIRINATHAN/vuln-ner-bert")