文章/答案/技术大牛

发布

社区首页 >问答首页 >如何在DeepPavlov分类器中获得概率？

问如何在DeepPavlov分类器中获得概率？
EN

Stack Overflow用户

提问于 2019-09-27 19:29:44

回答 1查看 200关注 0票数 2

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

from deeppavlov import configs, train_model

model = train_model(configs.classifiers.intents_snips)

model(['Some sentence'])

我想要的输出像np.array的类长度的数量，但目前的输出是一个标签，如“播放音乐”。

发布于 2019-11-14 21:49:24

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

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

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

要获取类索引，可以运行以下命令

dict(model['classes_vocab'])

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/58133683

复制

相似问题

问如何在DeepPavlov分类器中获得概率？EN