Skip to content

TensorFlow implementation of "Generating Sentences from a Continuous Space"

License

Notifications You must be signed in to change notification settings

zhangyijia1979/original_textvae

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TensorFlow implementation of "Generating Sentences from a Continuous Space"

paper:https://arxiv.org/abs/1511.06349

This is NOT an original implementation. There may be some minor differences from the original structure.

Results are reported in my blog

Prerequisites

  • Python 3.5
  • tensorflow-gpu==1.3.0
  • matplotlib==2.0.2
  • numpy==1.13.1
  • scikit-learn==0.19.0

Preparation

Dataset is not contained. Please prepare your own dataset.

  • Sentence

Pickle file of Numpy array of word ids (shape=[batch_size, sentence_length]).

  • Dictionary

Pickle file of Python dictionary. It should contain "", "", "" as meta words.

  dictionary = {word1: id1,
                word2: id2,
                ...}

Usage

Train

  1. modify config.py
  2. run
  python train.py

Get sample sentences

  1. modify sampling.py
  2. run
  python sampling.py

License

MIT

Author

Ryo Kamoi

About

TensorFlow implementation of "Generating Sentences from a Continuous Space"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%