jonghkim / keras-seq2seq-models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

keras-seq2seq-models

Work-in-Progress

The purpose of this project is to explore different s2s models based on Keras Functional API

Data

Models

  • vanilla seq2seq model (Done)
  • seq2seq model with attention mechanism (Done)
  • seq2seq auto-encoder model with adversarial network and style embedding (Done)
  • pointer network model (Work-in-Progress)
  • pointer-generator model (Work-in-Progress)
  • transformer model (Work-in-Progress)
  • bert model (Work-in-Progress)

Usage

# For vanilla seq2seq model: Solve Translation Problem
python -m bin.seq2seq_model_train
python -m bin.seq2seq_model_test

# For seq2seq with attention mechanism model: Solve Translation Problem
python -m bin.seq2seq_attn_model_train
python -m bin.seq2seq_attn_model_test

# For seq2seq auto-encoder model with adversarial network and style embedding: Solve Style Transfer Problem
python -m bin.seq2seq_adv_style_model_train
python -m bin.seq2seq_adv_style_model_test

# For pointer network model (*Work-in-Progress*): Solve Interger Sequence Ordering Problem
python -m bin.ptr_network_model_train
python -m bin.ptr_network_model_test

Code References

Paper References

For Citation

@misc{jonghkim,
  author = {Jongho Kim},
  title = {keras-seq2seq-models},
  year = {2018},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/jonghkim/keras-seq2seq-model}},
}

About


Languages

Language:Python 100.0%