我正在使用 rb-libsvm 进行预测
x.each_with_index do |e, i|
predictions << model_w2v.predict_probability(e).to_i
end
返回一个数组,其中第一个元素是标签,第二个元素是所有标签的概率。例如。 [2.0, [0.0, 0.0, 0.0, 0.0, 0.0]]
。每次标签都不同,但概率始终为零。我想知道是否是因为从 c 转换为 ruby 时数字太小并且四舍五入为 0.0。这是the relevant c code :
estimates = rb_ary_new();
for (i = 0; i < model->nr_class; i++)
rb_ary_push(estimates, rx_from_double(c_estimates[i]));
如何缓解这种情况?
最佳答案
rb-libsvm
gem 和底层 libsvm
需要将概率参数设置为 true。因此,我只需添加这一行并重新训练模型:
parameter.probability = 1
其中参数
之前定义为
parameter = Libsvm::SvmParameter.new
关于ruby - rb-libsvm 中的零概率,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35115974/