代码之家  ›  专栏  ›  技术社区  ›  yurnero

用MATLAB glmval预测二进制响应?

  •  0
  • yurnero  · 技术社区  · 6 年前

    (我的设置与这篇文章非常相似 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。

    0 回复  |  直到 6 年前
    推荐文章