masonzmz / ESTD

Empathy Style Transformer with Discriminative mechanism

ESTD: Empathy Style Transformer with Discriminative mechanism

[ADMA 2022] Mingzhe Zhang, Lin Yue, Miao Xu*

Overview

we employ ESTD to transfer a sentence from the source to the target text(higher empathy level). We believe our method and findings are a crucial step in establishing a friendly and inclusive online communication environment while furthering the development of a mental health platform.

Set-up

Operation System:

Requirements:

GPU:

Environment

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
pip install transformers
pip install datasets

Preparing Data

from datasets import load_dataset
dataset = load_dataset("blended_skill_talk")

Evaluation

Evaluation Code

python eval.py --gpu 0 --modelpath [model_path] --model ESTD

Training

python /src/run_training.py