PyTorch Beam Search Decoding

PyTorch ビーム検索

https://github.com/shawnwun/NNDIAL に基づく seq2seq モデルのビーム検索デコードの PyTorch 実装。デコードは文ごとに個別に行われ、ノードを優先順位付きキューに保存します。

使用法: BeamSearchNode.eval を通じて、デコードに対する追加の報酬を指定できます。アテンションの有無にかかわらずモデルに機能します。