目前實現了基於tensorflow的支持的帶attention的seq2seq。基於tf 1.0官網contrib路徑下seq2seq
由於后續版本不再支持attention,遷移到melt並做了進一步開發,支持完全ingraph的beam search(更快速)
以及outgraph的交互式beam search(更靈活),其中ougraph的beam search支持alignments的輸出。
attention的可視化也就是alignments的展示如下圖(輸入句子預測用戶可能的搜索詞):
Out graph beam search
In graph beam search
Calc text to text prob (計算文本到文本的生成概率)