File size: 1,949 Bytes
7194f39 374d1fb 7194f39 374d1fb 7194f39 374d1fb 7194f39 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
---
language: en
pipeline_tag: text-classification
library_name: transformers
tags:
- text-classification
- emotional-support
- empathy
- mental-health
license: mit
datasets:
- esconv
---
# Emotional Support Strategy Classifier
This model is a fine-tuned RoBERTa-base model for classifying emotional support conversation strategies.
## Model Description
- **Base Model**: roberta-base
- **Task**: Multi-class text classification
- **Training Data**: ESConv (Emotional Support Conversation) dataset
- **Number of Labels**: 8
## Labels
The model classifies text into 8 emotional support strategies:
0. Affirmation and Reassurance
1. Information
2. Others
3. Providing Suggestions
4. Question
5. Reflection of feelings
6. Restatement or Paraphrasing
7. Self-disclosure
## Usage
```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
# Load model and tokenizer
model_name = "RyanDDD/empathy-strategy-classifier"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
# Example prediction
text = "I understand how you feel. It's completely normal to feel this way."
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
outputs = model(**inputs)
predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
predicted_class = torch.argmax(predictions, dim=-1).item()
print(f"Predicted strategy: {model.config.id2label[predicted_class]}")
```
## Training
Fine-tuned on the ESConv dataset using the Hugging Face Transformers library.
## Citation
If you use this model, please cite the ESConv dataset:
```bibtex
@inproceedings{liu2021towards,
title={Towards Emotional Support Dialog Systems},
author={Liu, Siyang and Zheng, Chujie and Demasi, Orianna and Sabour, Sahand and Li, Yu and Yu, Zhou and Jiang, Yong and Huang, Minlie},
booktitle={Proceedings of ACL},
year={2021}
}
```
|