led灯|5分钟NLP:Python文本生成的Beam Search解码( 二 )


                                       input_len=len(input_ids[0
))
print(tokenizer.decode(beam_search_output[0
))
print(f\"\log_prob: {beam_search_log_prob:.2f\")
输出如下:

分时和连贯性要比贪婪的方法好很多 , 对吧 。
https://avoid.overfit.cn/post/ba2eb47bb35d43d99fb58333d37f13cb
作者:Jason LZP