Aikaksh-Singh-Routela/cybersecurity-bert-model

Cybersecurity BERT Classifier

This model is a fine-tuned bert-base-uncased model that classifies cybersecurity alerts into five threat categories.

Model Details

• —Base Model: bert-base-uncased
• —Task: Text Classification
• —Labels:
• —0: Ransomware
• —1: DDoS
• —2: Insider Threat
• —3: Web Attack
• —4: Benign

Intended Uses & Limitations

This model is intended for security operations center (SOC) teams to automatically triage and classify security alert text. It achieves 92.86% accuracy on a curated test set.

How to Use

You can use this model directly with the Transformers pipeline for text classification:

from transformers import pipeline

classifier = pipeline("text-classification", model="Aikaksh-Singh-Routela/cybersecurity-bert-model")

result = classifier("Files encrypted with ransom demand for Bitcoin payment")
print(result)
# Expected output: [{'label': 'Ransomware', 'score': 0.9286}]