Skip to content

demon386/tf_bucket_seq2seq

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Bucket-based Tensorflow seq2seq model

This is a demo of Bucket-based Tensorflow seq2seq model, with supporting for both training and decoding.

Please see test_predict in test_bucketmodel.py for usage.

Internally, it uses tf.contrib.legacy_seq2seq.model_with_buckets to build multiple models for each bucket (all share the same set of parameters). It can handle data pre-process and data post-process for using bucket-based models.

The test can be runned with pytest after you installed pytest package (pip install pytest).

It has been tested under Tensorflow docker image, with TF version 1.0.1

About

Bucket-based Tensorflow seq2seq model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages