r - 値をrの因子に分類する方法

Question

ロジスティック回帰を使用して y の確率を取得しました。次のことを行いました。

fit.model <- glm (y~ x1 +x2 , data = mydata, family=binomial)  
pred_model<- plogis(predict(fit.model, mydata))

ここで、カットオフ値 0.5 を使用して確率を分類したいと思います。
これを試しましたが、うまくいかない可能性があります。

class <- ifelse(pred_model>0.5, "yes" , "no" )       

助言がありますか？

Answer 1

これはうまくいくはずです：

class <- factor(ifelse(pred_model>0.5, "yes", "no"), c("yes", "no"))