search

xmxoxo / BERT-train2deploy

BERT模型从训练到部署
517 stars 165 forks source link

请问，如果想输出每个标签对应的score，应该怎么修改呢 #20

Open chentutu opened 4 years ago

lc222 commented 4 years ago

我现在也遇到这个问题，请问解决了吗

LeoWood commented 4 years ago

修改bert_base安装目录中的..\Lib\site-packages\bert_base\server__init__.py文件： line 465： return EstimatorSpec(mode=mode, predictions={ 'client_id': features['client_id'], 'encodes': tf.argmax(pred_probs[0], axis=-1), 'score': tf.reduce_max(pred_probs[0], axis=-1) }) 修改为： return EstimatorSpec(mode=mode, predictions={ 'client_id': features['client_id'], 'encodes': pred_probs[0], 'score': pred_probs[0] }) line 515: pred_label_result = [self.id2label.get(x, -1) for x in r['encodes']] 修改为： pred_label_result = r['encodes'].tolist()