PaddleHub 请问语义匹配预测结果能否输出匹配的概率得分,而不仅仅是0和1?

q9yhzks0  于 5个月前  发布在  其他
关注(0)|答案(2)|浏览(102)

请问预测结果能否输出匹配的概率得分,而不仅仅是0和1?

将predict接口参数return_result=False,按照如下代码进行后处理:

run_states = pointwise_matching_task.predict(
    data=text_pairs,
    max_seq_len=args.max_seq_len,
    label_list=dataset.get_labels(),
    return_result=False,
    accelerate_mode=False)

for batch_states in run_states:
    # 对应label概率值
    batch_results = batch_states.run_results
    print(batch_results)

@Steffy-zxf 你好 我按照上面的方式,输出只有0和1,帮忙看下

  • Originally posted by @dickobe in #868 (comment)*
kadbb459

kadbb4591#

此问题目前需要在 paddlehub/finetune/task/matching_task.py 文件中,在第750行进行修改后修复:

if return_result or not self._compatible_mode:
    return self._postprocessing(run_states)

修改为:

if return_result:
    return self._postprocessing(run_states)

1.8版本中的加入的兼容性代码影响了predidt结果处理的逻辑,之后2.0版本更新支持动态图版本的pointwise_matching_task,将修复这个问题。

jgwigjjp

jgwigjjp2#

遇到和你一样的问题了,貌似只有改源码了。

相关问题