Skip to content
/ ESTD Public

Empathy Style Transformer with Discriminative mechanism

License

Notifications You must be signed in to change notification settings

masonzmz/ESTD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

83 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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.

structure

Set-up

Operation System:

macOS Badge Linux Badge Ubuntu Badge

Requirements:

Python PyTorch Scikit-learn NumPy tqdm pandas SciPy colorama cudatoolkit datasets matplotlib nltk tokenizers transformers seaborn

GPU:

Nvidia

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

Releases

No releases published

Packages

No packages published

Languages