Aikaksh-Singh-Routela's picture
Update README.md
9d4401e verified
---
language: en
license: mit
tags:
- sentiment-analysis
- cybersecurity
- bert
widget:
- text: "Security team successfully prevents major data breach"
example_title: "Positive"
- text: "Ransomware attack shuts down hospital systems"
example_title: "Negative"
- text: "New security guidelines published by agency"
example_title: "Neutral"
---
# Security Sentiment Analyzer
Fine-tuned BERT model that analyzes sentiment of cybersecurity news headlines.
## Model Details
- **Base Model:** `bert-base-uncased`
- **Task:** Text Classification (Sentiment Analysis)
- **Labels:**
- `0`: Positive 🟢
- `1`: Negative 🔴
- `2`: Neutral 🔵
## Accuracy
**87.5%** on test set
## How to Use
```python
from transformers import pipeline
classifier = pipeline("text-classification", model="Aikaksh-Singh-Routela/security-sentiment-model")
result = classifier("Ransomware attack shuts down hospital systems")
print(result)