Рисунок трехмерной границы решения логистической регрессии

Question

Я подобрал модель логистической регрессии, учитывающую 3 переменные. Я хотел бы сделать 3D-график данных и нарисовать границу решения (что, я полагаю, будет плоскостью здесь).

Я нашел онлайн пример, который относится к случаю (так что вы можете загрузить данные непосредственно)

mydata <- read.csv("http://www.ats.ucla.edu/stat/data/binary.csv") 
mylogit <- glm(admit ~ gre + gpa + rank, data = mydata, family = "binomial") 

Я думал об использовании пакета 3Dscatterplot, но я не уверен, что уравнение я должен написать для рисования границы. Есть идеи?

Большого спасибо,

Answer 1

Граница решения будет 3-d самолетом, который можно построить с любым 3-ем черчением пакета в R. Я буду использовать persp путем определения х сетки, а затем вычисления соответствующее значение г с функцией outer:

# Use iris dataset for example logistic regression 
data(iris) 
iris$long <- as.numeric(iris$Sepal.Length > 6) 
mod <- glm(long~Sepal.Width+Petal.Length+Petal.Width, data=iris, family="binomial") 

# Plot 50% decision boundary; another cutoff can be achieved by changing the intercept term 
x <- seq(2, 5, by=.1) 
y <- seq(1, 7, by=.1) 
z <- outer(x, y, function(x, y) (-coef(mod)[1] - coef(mod)[2]*x - coef(mod)[3]*y)/
     coef(mod)[4]) 
persp(x, y, z, col="lightblue")