Vowpal Wabbit python包装空预测文件

时间:2016-11-14 00:08:43

标签: vowpalwabbit

从pyvw包装器调用时的预测文件为空。 例如,我正在做类似的事情,

vw = pyvw.vw(" -i cb.model --cb_explore 50 --cover 10 -p prediction.txt")
ex = vw.example(" | label label2")
vw.predict(ex)
vw.finish()
ex.finish() 

这会创建prediction.txt但不会向其写入任何内容。

我非常感谢任何指导。

谢谢!

1 个答案:

答案 0 :(得分:1)

此代码段可能会对您有所帮助:

from vowpalwabbit import pyvw

def to_vw(clf, text, str_label):
    vw_example = str('{} |f {} '.format(str_label, text))
    return clf.example(vw_example)

clf = vw = pyvw.vw(
    loss_function='logistic', oaa=2,
    link='logistic', raw_predictions='output.txt'
)

ex = to_vw(clf, 'I like vowpal wabbit. But not that much.', '1')
clf.learn(ex)
clf.predict(ex, labelType=pyvw.pylibvw.vw.lMulticlass)

您应该将概率写在output.txt文件中。