我从许多资源中读到,您可以通过设置
来检索预测概率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]
有可能吗?