Контур модели KNN в ggplot?

У меня есть модель KNN, которую я рисую с использованием функции contour. Это простой пример такого рода вещи, которые я делаю (на основе this Purdue exam):Контур модели KNN в ggplot?

library(class) 
library(nnet) 

TrainC<-read.table("http://miner.chem.purdue.edu/Exam1/TrainC.dat") 
names(TrainC)<-c("x1","x2","y") 

K=15 

p <- as.matrix(TrainC[, -3]) 
xp <- seq(min(TrainC$x1), max(TrainC$x1), length = 50); np <- length(xp) 
yp <- seq(min(TrainC$x2), max(TrainC$x2), length = 50) 
tp<-TrainC$y 

yhat <- knn(p, p, tp, k = K) 

plot(TrainC[, 1], TrainC[, 2], xlab = "x1", ylab = "x2", col=as.numeric(TrainC$y)+1) 
pt <- expand.grid(x1 = xp, x2 = yp) 
Z <- knn(p, pt, tp, k = K) 
zp<-class.ind(Z)[,1] - class.ind(Z)[,2] 

contour(xp, yp, matrix(zp, np), add = T, levels = 0, labex = 0)

Мой вопрос: как я могу сделать этот же сюжет в ggplot? В частности, как я могу сделать эквивалент contour?

источник

2011-01-19 griffin

В вашем коде отсутствует имя пакета, из которого вы получили функцию «class.ind», и даже если я пройду эту ошибку, я не вижу, где определено np. –

@DWin Извините ... class.ind is from 'nnet'; это исправлено. np определяется в строке, где xp определяется (после точки с запятой). – griffin

d <- transform(melt(matrix(zp, np)), xp=xp[X1], yp=yp[X2]) 
ggplot(d, aes(xp, yp, z=value)) + 
    geom_contour() + 
    geom_point(aes(x1, x2, colour=y, z=NULL), data=TrainC)

источник

2011-01-19 05:06:02 kohske

После кода, попробуйте следующее:

library(ggplot2) 

p <- ggplot(TrainC, aes(x1, x2)) 
p + geom_point(aes(colour = as.numeric(y) + 1)) 

df <- data.frame(x = rep(xp, np), y = rep(yp, each = np), z = zp) 
p <- ggplot(df, aes(x, y, z = z)) 
p + stat_contour(bins = 1)

Это два отдельных участка. Не знаете, как их объединить, но в документации here вам стоит начать.

источник

2011-01-19 05:01:08

Вы можете просто взять свое первое изображение и добавить 'p + stat_contour (data = df, aes (x, y, z = z), bins = 1)'. – Shane

Контур модели KNN в ggplot?

ответ

Смежные вопросы