Modelpublic

RobroKools/vad-bert

sourceHugging Facemitupdated 1y agoView on Hugging Face

5likes7kdownloads

Model Card

vad-bert

A BERT-based model fine-tuned to predict Valence, Arousal, and Dominance (VAD) values from text.

Intended use

This model is intended for regression tasks on emotional dimensions. It outputs 3 float values corresponding to:

—Valence (pleasant vs unpleasant)
—Arousal (calm vs excited)
—Dominance (controlled vs in control)

Example

python

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("RobroKools/vad-bert")
model = AutoModelForSequenceClassification.from_pretrained("RobroKools/vad-bert")

inputs = tokenizer("I'm feeling great!", return_tensors="pt")
outputs = model(**inputs)

vad = outputs.logits.detach().squeeze().tolist()
print(vad)  # [valence, arousal, dominance]