SecureBERT: A Domain-Specific Language Model for Cybersecurity
Paper • 2204.02685 • Published • 1
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("CyberPeace-Institute/SecureBERT-NER")
model = AutoModelForTokenClassification.from_pretrained("CyberPeace-Institute/SecureBERT-NER")Quick Links
Named Entity Recognition for Cybersecurity
This model has been finetuned with SecureBERT (https://arxiv.org/abs/2204.02685) on the APTNER dataset (https://ieeexplore.ieee.org/document/9776031)
NER Classes
- Downloads last month
- 304
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="CyberPeace-Institute/SecureBERT-NER")