关于kw_mask部分

[Htter]

首先谢谢作者开放代码~

看了您的源码，发现create_attention_mask_from_keyword_mask函数生成的kw_mask只有A's token到B's key word的映射，按这部分的注释似乎是包含A到B以及B到A两方的token 及 kw映射，代码似乎搞错了

[DataTerminatorX]

有的， modeling.py 的 line 693~694

[zhx970928]

你好，我想请问下你跑出来的数据格式是怎么样的？然后输入数据是怎么样的？你的关键词用的是源代码提供的方法吗？ 使源代码产生的数据是prediction[0.5555，0.4445]，类似这种格式，他代表了什么？如果prediction[0]代表标签为0的概率，如果prediction[1]代表标签为1的概率，但是源代码跑出来的数据都是prediction[0]>prediction[1],那模型预测均为0，是为什么那？