Skip to content

lydiahjchung/TextStyleTransfer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text Style Transfer

SWCON401-00 Capstone Design

Generating sarcastic utterances via text style transfer. The original SARC dataset with instructions can be found here.

Using pre-trained BART

Referenced KoBART Chit-Chat model to implemt. Referenced code can be found here.

Training

mkdir logs
python bart/main.py --gradient_clip_val 1.0 --max_epochs 10 --default_root_dir logs --gpus 1

Using Cross-aligned Auto Encoder

Model proposed in Style Transfer from Non-Parallel Text by Cross-Alignment, NIPS, 2017.

Training

mkdir save
python cross-aligned/style_transfer.py --saves_path ./save --train ./sarc/sarc.train --max_epochs 120 --vocab ./tmp/sarc.vocab

Using Style Transformer

Model proposed in Style Transformer: Unpaired Text Style Transfer without Disentangled Latent Representation, ACL, 2019.

Training

mkdir save
python main.py

About

SWCON401-00 Capstone Design

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published