我正在尝试实施一种方法来评估预测概率，就像借助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
最佳答案
我对您使用的特定代码了解不多。但是对于分类（我想这就是您要寻找的），BERT在预训练的BERT的最后一层上使用了线性+ soft-max层。因此，计算概率应该很简单。我显然找到了an example with ktrain，您可以在其中指定所需的概率（第23行）。
关于python - BERT分类器中每个标签的预测概率，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/59265253/