如何使用PaddleHub model.predict()方法输出概率值？
收藏
如何使用PaddleHub model.predict()方法输出概率值？
目前输出的是label_map = {0: 'similar', 1: 'dissimilar'}中的某个类别，我想要输出概率值，要如何调用呢？
data = [
['这个表情叫什么', '这个猫的表情叫什么'],
]
label_map = {0: 'similar', 1: 'dissimilar'}
model = hub.Module(
name='ernie_tiny',
version='2.0.2',
task='text-matching',
label_map=False)
results = model.predict(data, max_seq_len=128, batch_size=1, use_gpu=False)
print(results)
for idx, texts in enumerate(data):
print('TextA: {}\tTextB: {}\t Label: {}'.format(texts[0], texts[1], results[idx]))
东君
已解决
10#
回复于2022-08
results = model.predict(data, max_seq_len=20, batch_size=len(label), use_gpu=False, return_prob = True) 改个参数。
0
收藏
请登录后评论
解决了，修改predict方法，输出里面的prob即可
可以设自己这条为解决答案哦（/偷笑）
你好，请问如何修改函数的
请问你是修改的源码还是重写的方法
请问是怎么修改呢，能请教一下吗
各位可以提供一下修改方式吗
results = model.predict(data, max_seq_len=20, batch_size=len(label), use_gpu=False, return_prob = True)
改个参数。