textmatch(文本匹配)模型輸出score卡閾值一個細節

最後輸出結果是[batch_size,2]
但不是通過[:,0][:,1]的大小相比得出結果,
也不是改成輸出[batch_size,1]
而是[batch_size,2][:,1]然後卡閾值。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章