predict的BUG,i fix it already(那是不可能的)。
Created by: yangtianyu92
作为管理学院唯一指定代码工具人,你们第一次一度让我感觉自己像个HACKER。
PaddleHub 0.5版本
nlp_reader.py
LACClassifyReader
'''
if self.label_map:
if example.label not in self.label_map:
raise KeyError(
"example.label = {%s} not in label" % example.label)
label_id = self.label_map[example.label]
else:
label_id = example.label
'''
log打出label_map = {'risk4': 4, 'risk2': 2, 'risk3': 3, 'risk0': 0, 'risk1': 1} 而我输入文本 “阿拉伯7国发布通报修订冷冻龙虾的相关规定”, 类别应属于risk4 但是这时候example.label输出都为0,按照这段代码的逻辑就会raise an error。
修复方法就是把这段注释掉,直接label_id = example.label