如何在深度巴甫洛夫分类器中获得概率?

Question

我使用 DeepPavlov 训练我的分类器，然后当我为某个示例函数调用训练模型时，只返回一个类标签，但我想获取每个类的概率。我没有找到可以让我获得概率的函数参数。 有没有人遇到过这样的问题？谢谢！

from deeppavlov import configs, train_model
model = train_model(configs.classifiers.intents_snips)
model(['Some sentence'])

我想要像 np.array 这样的输出，具有类长度的数量，但当前输出是一个标签，如 ['PlayMusic']。

Answer 1

您可以在推断之前将配置chainer.out参数更改为["y_pred_probas"]，但如果您想在同一配置上训练模型，它也很可能要求您更新更改train.metrics。

或者，您可以像这样调用模型

model.compute(['Some sentence'], targets=["y_pred_probas"])

要获取类索引，您可以运行

dict(model['classes_vocab'])