ggplot中KNN模型的轮廓？

我有一个KNN模型，我正在使用contour函数进行绘图。这是我做那种事（基于this Purdue exam）的一个简单的例子：ggplot中KNN模型的轮廓？

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来自'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

ggplot中KNN模型的轮廓？

回答

相关问题