python - BERT分类器中每个标签的预测概率

Question

我正在尝试实现一种评估预测概率的方法，就像在 scikit learn 的帮助下完成的

confidence = model._predict_proba_lr(x_count).max() * 100

有没有办法使用 BERT 模型来评估相同的结果？

目前使用 Bert Base Uncased。
使用Ktrain库(内部使用Keras)
引用代码:https://github.com/amaiya/ktrain/blob/master/examples/text/20newsgroups-BERT.ipynb

Answer 1

在ktrain中，您可以将return_proba=True传递给predictor.predict方法来输出概率。请参阅text classification tutorial了解更多详情。