|
--- |
|
language: |
|
- en |
|
pipeline_tag: conversational |
|
tags: |
|
- psychology |
|
- dialogues |
|
- empathy |
|
- gpt2 |
|
--- |
|
|
|
## Training data |
|
It was trained on a large corpus of text, including some emotionally engaging datasets such as the "Facebook Empathetic Dialogues" dataset containing 25k conversations. |
|
A dataset of 25k conversations grounded in emotional situations to facilitate training and evaluating dialogue systems. |
|
You can find a dataset [here](https://www.kaggle.com/datasets/atharvjairath/empathetic-dialogues-facebook-ai). |
|
```python |
|
>>> from transformers import AutoTokenizer, AutoModelForCausalLM |
|
>>> tokenizer = AutoTokenizer.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues") |
|
>>> model = AutoModelForCausalLM.from_pretrained("AliiaR/DialoGPT-medium-empathetic-dialogues") |
|
``` |
