如何从Vowpal Wabbit [Python Package]获得预测概率?

时间:2016-09-06 16:58:03

标签: python machine-learning nlp vowpalwabbit

我从许多资源中读到,您可以通过设置

来检索预测概率
oss_function='logistic'
probabilities=True
oaa=N

好的,我已经做到了。我的剧本:

from vowpalwabbit.sklearn_vw import VW

X = ['1 | feature1:2.5',
     '2 | feature1:0.11 feature2:-0.0741',
     '3 | feature3:2.33 feature4:0.8 feature5:-3.1',
     '1 | feature2:-0.028 feature1:4.43',
     '2 | feature5:1.532 feature6:-3.2']

model = VW(
    convert_to_vw=False, loss_function='logistic', oaa=3, probabilities=True
)

model.fit(X)

print model.predict([' | feature1:0.11 feature2:-0.0741'])

此代码段输出如下内容:

[ 28036672.]

但是,我喜欢每个类的概率,作为softmax图层输出:

[0.3, 0.6, 0.1]

有可能吗?

0 个答案:

没有答案