请问预测结果能否输出匹配的概率得分,而不仅仅是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)*
2条答案
按热度按时间kadbb4591#
此问题目前需要在
paddlehub/finetune/task/matching_task.py
文件中,在第750行进行修改后修复:修改为:
1.8版本中的加入的兼容性代码影响了predidt结果处理的逻辑,之后2.0版本更新支持动态图版本的pointwise_matching_task,将修复这个问题。
jgwigjjp2#
遇到和你一样的问题了,貌似只有改源码了。