Skip to content

nlpofwhat/TransformerForTopic-to-essay

Repository files navigation

TransformerForTopic-to-essay

transformer for topic-to-essay

input : no more than 5 tokens, output : a sequence

main reqirements

python 3.5.3 tensorflow 1.12.0 cuda 9.0 cudnn 7.4

The data is put in the 'data/', you can train dircetly. python train.py

Attention! because the input is unorderd, I drop the position embedding. Bleu score and dist-n score is emplyed to test.

reference: https://github.com/Kyubyong/transformer

About

TopicToEssay

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages