Disaster BERT — Fine-Tuned 12-Class Disaster Text Classification Model

Fine-tuned bert-base-uncased model trained on 203,976 disaster news articles to classify natural and man-made emergency occurrences into 12 distinct hazard categories.

Model Performance & Evaluation Metrics

Test Accuracy: 89.31%
Test Macro F1-Score: 84.66%
Test Weighted F1-Score: 89.12%
Epochs Trained: 4
Training Samples: 203,976

Python Usage Example

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

model_id = "harry2708/disaster-bert"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)

text = "Flood situation in Assam improves marginally as river water recedes"
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)

with torch.no_grad():
    outputs = model(**inputs)
    probs = torch.softmax(outputs.logits, dim=1).squeeze(0)
    top_idx = torch.argmax(probs).item()
    confidence = probs[top_idx].item()

predicted_label = model.config.id2label[top_idx]
print(f"Predicted Disaster: {predicted_label} ({confidence * 100:.2f}% Confidence)")

harry2708/disaster-bert

Disaster BERT — Fine-Tuned 12-Class Disaster Text Classification Model

Model Performance & Evaluation Metrics

12 Supported Disaster Categories

Python Usage Example