可視化展示attention(seq2seq with attention in tensorflow)


目前實現了基於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 (計算文本到文本的生成概率)

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM