我如何获得DeepPavlov分类器中的概率？

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'])