Skip to content

An improved implementation of Beam Search Decoding in RNN-based Seq2Seq Architecture

License

Notifications You must be signed in to change notification settings

dinhngoc267/beam_search_seq2seq

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 

Repository files navigation

Seq2Seq Beam Search Decoding for Pointer Generator

This code is improved from jojonki's repository

The idea of his approach is nice which uses Node and Linked List structure.

But the implementation of Beam Search seems not correct:

  • About definition of a step
  • And finding end nodes is not correct

I fixed a little bit in his code. And the performance of my Pointer Generator is increased from 3 to 5 points bleu score.

My code can be modified to use in any RNN-based Seq2Seq architecture

Reference:

About

An improved implementation of Beam Search Decoding in RNN-based Seq2Seq Architecture

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages