【问题标题】:How i can get probabilities in DeepPavlov classifier?我如何在 DeepPavlov 分类器中获得概率?
【发布时间】:2019-09-27 11:29:43
【问题描述】:

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

from deeppavlov import configs, train_model

model = train_model(configs.classifiers.intents_snips)

model(['Some sentence'])


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

【问题讨论】:

标签: python machine-learning deep-learning deeppavlov


【解决方案1】:

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

或者你可以调用你的模型

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

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

dict(model['classes_vocab'])

【讨论】:

    猜你喜欢
    • 2018-08-28
    • 2019-10-25
    • 2020-10-09
    • 2015-10-28
    • 2016-06-28
    • 2016-02-13
    • 2017-06-12
    • 2011-09-22
    • 1970-01-01
    相关资源
    最近更新 更多