python - 我如何在 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'])