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