/original_textvae

TensorFlow implementation of "Generating Sentences from a Continuous Space"

Primary LanguagePythonMIT LicenseMIT

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.

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 "<EOS>", "<PAD>", "<GO>" 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