使用数据集iris, 验证Petal.Length, Petal.Width两个特征的分类能力。代码如下:
with(iris, plot(Petal.Length, Petal.Width, col=as.integer(Species)))
text(2.2, 0.3, "setosa")
text(3.0, 1.3, "versicolor")
text(6.5, 1.7, "virginica")
效果如下,可以看出,Petal.Length, Petal.Width两个特征的分类能力很强:
调用kknn函数进行模型训练与预测,代码如下:
library(kknn)
data(iris)
m <- dim(iris)[1] #获取数据集记录条数
val <- sample(1:m, size =round(m/3), replace = FALSE, prob= rep(1/m, m)) #抽样,选取三分之二的数据作为训练集。
iris.learn <- iris[-val,] #选取训练集
iris.valid <- iris[val,] #选取验证集
#训练模型并进行预测分类
iris.kknn <- kknn(Species~Petal.Length + Petal.Width,iris.learn, iris.valid, k=7, distance=2)
summary(iris.kknn)