bert do_predict返回概率值

bert 的tensorflow 版本做 do_predict=TRUE 时，会产生一个文件 test_results.tsv 文件，对输入的每一个句子生成一个概率值。
由于我做的是双文本的相似度计算，更想得到每一对文本的相似度
请问是否有方法采用这个 .tsv 这个文件得到相似度呢，或者返回二分类？或者有更好的方式能返回bert的相似度？
感谢各位的解答。

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 以下回答来自chatgpt:

  抱歉，由于问题并未给出，请您补充问题后再进行提问。

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^