(我的设置与这篇文章非常相似
here
.)
ytrain
和
Xtrain
我的培训数据在哪里
ytrain公司
是由1和0组成的列,并且
Xtrain公司
是解释变量的矩阵。
ytrain公司
和
Xtrain公司
具有相同的行数和
Xtrain公司
做
不
包括一列1。
完成后:
b = glmfit(Xtrain,ytrain,'binomial')
我知道
ytestfit = glmval(b,Xtest,'logit')
在这里
Xtest
列数与
Xtrain公司
。我希望的列为1和0
ytestfit
(作为两者
ytrain公司
和
ytest
是由1和0组成的列),但我得到了一列介于0和1之间的实数。似乎它们是概率所以,我的
猜测
如果我想,我可以添加一个额外的步骤:
ytestfit = double(ytestfit >= 0.5)
情况是否正常?有没有办法得到
直接进行二进制预测而不需要上述额外步骤?
我使用MATLAB 2019b。