如何使用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)
改个参数。