def get_label(sentence):
"""
Prediction of the sentence's label.
"""
feature = get_feature_test(sentence)
fd = {MODEL.albert.input_ids: [feature[0]],
MODEL.albert.input_masks: [feature[1]],
MODEL.albert.segment_ids:[feature[2]],
}
prediction = MODEL.sess.run(MODEL.albert.predictions, feed_dict=fd)[0]
print(prediction)
r=[]
for i in range(len(prediction)):
if prediction[i]!=0.0:
r.append(id2label(i))
return r
#return [id2label(l) for l in np.where(prediction==1)[0] if l!=0]
训练结束后predict.py脚本获取不到标签问题
知乎上也有很多人说predict.py脚本获取到的标签为空，其实不是训练数据有问题或者轮次不够，作者的get_label 函数逻辑有一些小小的问题，我这里简单修改了一下，可以成功获取到标签，新的predict.py 的get_label 函数如下：