深度学习概率输出
字段名翻譯取出概率
import numpy as np
#topv 是概率,topi是對(duì)應(yīng)哪個(gè)翻譯
topv, topi = decoder_output.data.topk(1)
gl = np.exp(topv.cpu().item())
glhz.append(gl)
if topi.item() == EOS_token:
decoded_words.append(’’)
break
else:
decoded_words.append(self.output_lang.index2word[topi.item()])
decoder_input = topi.squeeze().detach()
from functools import reduce
glhz=glhz[0:-1]
total_gl=reduce(lambda x,y :x*y,glhz)
把結(jié)束字段去掉
數(shù)字識(shí)別,簡(jiǎn)單網(wǎng)絡(luò)取出概率
output = model(data)
total=output.sum()fenbu=output/totalfenbu=fenbu.tolist()maxprob=sorted(fenbu[0],reverse=True)maxprob=round(maxprob[0],2)pred = output.max(1, keepdim=True)[1]pred = pred.view(-1)pred=np.array(pred).tolist()pred=pred[0]
總結(jié)
- 上一篇: Brat序列标注工具小结
- 下一篇: albert速度