r - 预测()函数的类型参数

Question

type="class"函数中的type="response"和predict有什么区别？

例如：

predict(modelName, newdata=testData, type = "class")

和

predict(modelName, newdata=testData, type = "response")

Answer 1

响应为您提供数值结果，而类为您提供分配给该值的标签。

响应使您可以确定阈值。例如，

glm.fit = glm(Direction~., data=data, family = binomial, subset = train)
glm.probs = predict(glm.fit, test, type = "response")

在 glm.probs中，我们有一些介于0和1之间的数值。现在我们可以确定阈值，例如0.6。方向有两个可能的结果，向上或向下。

glm.pred = rep("Down",length(test))
glm.pred[glm.probs>.6] = "Up"