Skip to content

A conditional variational autoencoder (CVAE) for text

Notifications You must be signed in to change notification settings

iconix/pytorch-text-vae

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

64 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A partial reimplementation of "Generating Sentences From a Continuous Space", Bowman, Vilnis, Vinyals, Dai, Jozefowicz, Bengio (https://arxiv.org/abs/1511.06349).

Based on code from Kyle Kastner (@kastnerkyle) https://github.com/kastnerkyle/pytorch-text-vae, adapted to support the deephypebot project.


Changes in this detached fork:

  • Update compatibility to Python 3 and PyTorch 0.4
  • Add generate.py for sampling
  • Add special support for JSON reading and thought vector conditioning
  • Some code cleanup
  • Add setup.py for package support as pytorchtextvae
  • Train/test data split support

About

A conditional variational autoencoder (CVAE) for text

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages