This model predicts Valence, Arousal, and Dominance (VAD) values for disaster news headlines.
It was trained on the EmoBank dataset and fine-tuned on disaster news headlines.
Model Details
Architecture: RoBERTa-based transformer model with regression heads for VAD prediction
Training Data: EmoBank dataset
Application: Emotional analysis of disaster news headlines
Date: 2025-03-16
Usage
python
1from transformers import RobertaTokenizer, AutoModel
2import torch
34# Load model and tokenizer5tokenizer = RobertaTokenizer.from_pretrained("postgrammar/disaster-news-vad-model")6model = AutoModel.from_pretrained("postgrammar/disaster-news-vad-model")78# Prepare input9text ="Earthquake devastates coastal town, rescue efforts underway"10inputs = tokenizer(text, return_tensors="pt")1112# Get predictions13with torch.no_grad():14 outputs = model(**inputs)1516# Extract VAD values (first three values in the output tuple)17valence, arousal, dominance = outputs[0], outputs[1], outputs[2]1819print(f"Valence: {valence.item():.4f}, Arousal: {arousal.item():.4f}, Dominance: {dominance.item():.4f}")
Citation
If you use this model, please cite:
@misc{disaster-news-vad-model,
author = {Hamed Yaghoobian},
title = {Disaster News VAD Model},
year = {2025},
publisher = {Hugging Face},
howpublished = {\url{https://huggingface.co/postgrammar/disaster-news-vad-model}}
}