breeze
Results
1
issues of
breeze
训练结束后predict.py脚本获取不到标签问题 知乎上也有很多人说predict.py脚本获取到的标签为空，其实不是训练数据有问题或者轮次不够，作者的get_label 函数逻辑有一些小小的问题，我这里简单修改了一下，可以成功获取到标签，新的predict.py 的get_label 函数如下： ``` 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]...