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

[chentutu]

No description provided.

[lc222]

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

[LeoWood]

修改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()