Aikaksh-Singh-Routela's picture
Update README.md
9d4401e verified
metadata
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

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)